AcWing 4411. 三仙归洞 (简单模拟)

三个倒扣着的不透明小碗排成一排。

随机挑选一个小碗，将一个小球置于碗中。

然后进行 nn 次操作，编号 1∼n1∼n。

对于第 ii 次操作：

• 如果 imod2=1imod2=1，则操作内容为将位于中间的碗和位于左边的碗交换位置。
• 如果 imod2=0imod2=0，则操作内容为将位于中间的碗和位于右边的碗交换位置。

我们不妨用 0,1,20,1,2 来表示左、中、右三个位置。

nn 次操作全部完成以后，装有小球的碗位于位置 xx。

请你计算，所有操作开始前，装有小球的碗所在的初始位置。

输入格式

第一行，一个整数 nn。

第二行，一个整数 xx。

输出格式

输出一个 0∼20∼2 的整数，表示所有操作开始前，装有小球的碗所在的初始位置。

数据范围

前 66 个测试点满足 1≤n≤51≤n≤5。
所有测试点满足 1≤n≤2×1091≤n≤2×109，0≤x≤20≤x≤2。

输入样例1：

4
2

输出样例1：

1

输入样例2：

1
1

输出样例2：

0

代码如下：

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;
int n, x;
int a[3];

int main() {
   ios::sync_with_stdio(false);
   cout.tie(nullptr);

   memset(a, 0, sizeof a);
   cin >> n >> x;

   a[x] = 1;

   n %= 6;

   while(n) {
       if(n & 1)
            a[0] ^= a[1] ^= a[0] ^= a[1];
       else 
            a[1] ^= a[2] ^= a[1] ^= a[2];
       n --;
   }

   for(int i = 0; i < 3; i ++)
   if(a[i] == 1) cout << i;

   return 0;
}