[C++] 백준 1547번 공

3개의 컵에서 첫번째 컵에 공을 넣고 시작합니다.
2개의 컵의 위치를 N번 바꿨을때
최종적으로 공을 가지고 있는 컵을 출력하는 문제입니다.

swap함수를 만들어주어서 문제를 해결하였습니다.

 

#include <iostream>
using namespace std;

int n;
int x, y;
int ball[4] = { 0, 1, 0, 0 };

void swap(int x, int y)
{
	int temp;
	temp = ball[x];
	ball[x] = ball[y];
	ball[y] = temp;
}

int main(int argc, char *argv[])
{
	cin >> n;
	for (int i = 1; i <= n; ++i) {
		cin >> x >> y;
		swap(x, y);
	}

	for (int i = 1; i <= 3; ++i) {
		if (ball[i] == 1) {
			cout << i << endl;
		}
	}

	return 0;
}