AtCoder Beginner Contest 172 F - Unfair Nim
有 N ( 2 ≤ N ≤ 300 ) N(2\le N \le300) N(2≤N≤300)堆石子,其中第 i i i堆有 A i ( 1 ≤ A i ≤ 1 0 12 ) A_i(1\le A_i\le 10^{12}) Ai(1≤Ai≤1012)个石子,两人进行尼姆(Nim)游戏,问至少要将多少石子从第 1 1 1堆移到第 2 2 2堆(不可全部移走,即只能移 0 0 0 ~ A 1 − 1 A_1-1 A1−1个石子),才能使得先手必败。不存在可行解时输出 − 1 -1 −1。
Nim博弈的必败局面要求: A 1 ⊕ A 2 ⊕ A 3 ⊕ ⋯ ⊕ A N = 0 A_1\oplus A_2\oplus A_3\oplus\cdots\oplus A_N = 0 A1⊕A2⊕A3⊕⋯⊕AN=0
设 X = A 3 ⊕ A 4 ⊕ A 5 ⊕ ⋯ ⊕ A N X = A_3\oplus A_4\oplus A_5\oplus\cdots\oplus A_N X=A3⊕A4⊕A5⊕⋯⊕AN, S = A 1 + A 2 S = A_1+A_2 S=A1+A2,移动石子后的 A 1 A_1 A1和 A 2 A_2 A2分别为 a a a和 b b b
则可转化题意为:
求最大的 a ≤ A 1 a\le A_1 a≤A1,使得下式成立
{ a + b = S a ⊕ b = X \begin{cases}a+b=S\\a\oplus b=X\end{cases} {a+b=Sa⊕b=X
算术加法运算和逻辑异或运算是很难联立求解的,这时候需要一个很巧妙的转化公式(易证):
a + b = ( a ⊕ b ) + 2 × ( a ∧ b ) a+b=(a\oplus b)+2\times(a\land b) a+b=(a⊕b)+2×(a∧b)代入可得 a ∧ b = ( S − X ) / 2 a\land b = (S-X)/2 a∧b=(S−X)/2,记为 D D D
则题意可进一步转化为:
求最大的 a ≤ A 1 a\le A_1 a≤A1,使得下式成立
{ a ∧ b = D a ⊕ b = X \begin{cases}a\land b=D\\a\oplus b=X\end{cases} {a∧b=Da⊕b=X这样就转化为了两个逻辑运算的联立式,可以对每一个二进位进行单独考虑。
设 a , b , D , X a,b,D,X a,b,D,X某一对应二进制位分别为 a i , b i , D i , X i a_i,b_i,D_i,X_i ai,bi,Di,Xi;
首先考虑若 D i D_i Di为 1 1 1, 则一定有 a i = 1 , b i = 1 a_i=1,\;b_i=1 ai=1,bi=1,故 a a a的最小值为 D D D,即应当 令 a a a的答案初值为 D D D;
接下来为使得答案 a a a尽可能大,从二进制高位到低位进行遍历,考虑以下四种情况:
最后考虑有哪些情况输出 − 1 -1 −1(无解):
#include
#define lowbit(x) ((x)&(-(x)))
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
const int INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
const int maxn = 5e5 + 10;
int N;
LL A[500], S, X, D, ANS;
int main()
{
scanf("%d", &N);
for(int i = 1; i <= N; i++)
{
scanf("%lld", &A[i]);
if(i <= 2)
S += A[i];
else
X ^= A[i];
}
if(S - X < 0 || (S - X) % 2 != 0)
{
printf("-1\n");
return 0;
}
else
D = (S - X) / 2;
LL ANS = D;
for(int i = 60; i >= 0; i--)
{
int x = ((X >> i) & 1LL), d = ((D >> i) & 1LL);
if(d && x)
{
printf("-1\n");
return 0;
}
if(!d && x && (ANS | (1LL << i)) <= A[1])
ANS |= (1LL << i);
}
if(ANS == 0 || ANS > A[1])
printf("-1\n");
else
printf("%lld\n", A[1] - ANS);
return 0;
}