백준 10813 공 바꾸기

https://www.acmicpc.net/

Baekjoon Online Judge

[문제 해석]

1~N까지의 번호가 있는 공이 각 번호에 맞게 위치해있고, 이 위치를 2개씩 바꾸는걸 반복하는 문제이다.

공 개수 N이 주어지고, 공 위치를 바꿀 횟수 M이 주어진다.
두번째 줄부터는 두 공의 위치가 입력되면 각 위치에 있는 공을 서로 바꾸면 된다.
주의할점은 공의 위치가 주어지는것이지 공의 번호가 주어지는것이 아니다.
k라는 벡터를 만들어 0번 인덱스부터 N번 인덱스까지 각 인덱스 숫자를 값으로 할당한다. (1번째는 숫자 1이 있겠지)
다음 for문에서는 M번을 반복하며 공의 위치를 바꿀건데, a,b라는 변수에 각각 숫자을 입력받아 k[a],k[b]의 위치를 바꿔준다. (k[a],k[b]인덱스가 공의 위치와 같음)
배열,벡터의 두 원소의 위치를 바꾸는 함수로 swap함수를 사용한다.
출력으로는 " "공백을 포함해서 배열을 for문을 통해 출력해준다.

[정답 코드]

#include <iostream>
#include <math.h>
#include <algorithm>
#include <vector>
#include <string>
using namespace std;

int main()
{
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	int N, M; cin >> N >> M;
	vector<int> k;
	int a, b;

	for (int i = 0; i <= N; i++)
	{
		k.push_back(i);
	}

	for (int i = 0; i < M; i++)
	{
		cin >> a >> b;

		swap(k[a], k[b]);
	}

	for (int i = 1; i <= N; i++)
	{
		cout << k[i] << " ";
	}
}