AtCoder Beginner Contest 172 F - Unfair Nim(Nim博弈,位运算)

AtCoder Beginner Contest 172 F - Unfair Nim

题意:

N ( 2 ≤ N ≤ 300 ) N(2\le N \le300) N(2N300)堆石子,其中第 i i i堆有 A i ( 1 ≤ A i ≤ 1 0 12 ) A_i(1\le A_i\le 10^{12}) Ai(1Ai1012)个石子,两人进行尼姆(Nim)游戏,问至少要将多少石子从第 1 1 1堆移到第 2 2 2堆(不可全部移走,即只能移 0 0 0 ~ A 1 − 1 A_1-1 A11个石子),才能使得先手必败。不存在可行解时输出 − 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 A1A2A3AN=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=A3A4A5AN 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 aA1,使得下式成立

{ a + b = S a ⊕ b = X \begin{cases}a+b=S\\a\oplus b=X\end{cases} {a+b=Sab=X

算术加法运算和逻辑异或运算是很难联立求解的,这时候需要一个很巧妙的转化公式(易证):
a + b = ( a ⊕ b ) + 2 × ( a ∧ b ) a+b=(a\oplus b)+2\times(a\land b) a+b=(ab)+2×(ab)代入可得 a ∧ b = ( S − X ) / 2 a\land b = (S-X)/2 ab=(SX)/2,记为 D D D

题意可进一步转化为:
求最大的 a ≤ A 1 a\le A_1 aA1,使得下式成立

{ a ∧ b = D a ⊕ b = X \begin{cases}a\land b=D\\a\oplus b=X\end{cases} {ab=Dab=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. D i = 1 ,    X i = 1 D_i=1,\;X_i=1 Di=1,Xi=1
    该情况无解。
  2. D i = 1 ,    X i = 0 D_i=1,\;X_i=0 Di=1,Xi=0
    此时一定有 a i = 1 ,    b i = 1 a_i=1,\;b_i=1 ai=1,bi=1,因为 a a a的初值为 D D D,该条件已保证。
  3. D i = 0 ,    X i = 1 D_i=0,\;X_i=1 Di=0,Xi=1
    此时可以为 a i = 1 ,    b i = 0 a_i=1,\;b_i=0 ai=1,bi=0或者 a i = 0 ,    b i = 1 a_i=0,\;b_i=1 ai=0,bi=1,为使得 a a a尽可能大,若有 a + 2 i ≤ A 1 a+2^i\le A_1 a+2iA1,则令该位 a i = 1 a_i=1 ai=1
  4. D i = 0 ,    X i = 0 D_i=0,\;X_i=0 Di=0,Xi=0
    此时一定有 a i = 0 ,    b i = 0 a_i=0,\;b_i=0 ai=0,bi=0,令该位保持 a i = 0 a_i=0 ai=0即可。

最后考虑有哪些情况输出 − 1 -1 1(无解):

  1. 2 2 2无法整除 ( S − X ) (S-X) (SX),或者 S − X < 0 S-X<0 SX<0时( D < 0 D<0 D<0)时,无解;
  2. 当存在 D i = 1 ,    X i = 1 D_i=1,\;X_i=1 Di=1,Xi=1时,无解;
  3. D > A 1 D>A_1 D>A1时,因为 D D D a a a的最小取值,而又必须满足 a 1 ≤ A 1 a_1\le A_1 a1A1,故该情况无解;
  4. 当最后求得 a a a的最大值为 0 0 0时,说明至少要将 A 1 A_1 A1全部移给 A 2 A_2 A2,不合题意,无解。


代码:

#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;
}

你可能感兴趣的:(★水题之路,★数学)