链接:https://ac.nowcoder.com/acm/contest/549/I
来源:牛客网
小A也听说了取石子这个游戏,也决定和小B一起来玩这个游戏。总共有n堆石子,双方轮流取石子,每次都可以从任意一堆中取走任意数量的石子,但是不可以不取。规定谁先取完所有的石子就获胜。但是小A实在是太想赢了,所以在游戏开始之前,小A有一次机会,可以趁小B不注意的时候选择其中一堆石子拿走其中的k个,当然小A也可以选择不拿石子。小A先手。双方都会选择最优的策略,请问在这样的情况下小A有没有必胜的策略,如果有输出YES,否则就输出NO。
思路:Nim游戏的模板题 枚举每一堆,判断取走后能否使必胜(维护前缀异或L数组和后缀异或R数组) 判断 L[i - 1] ^ x ^ R[i + 1] != 0(必胜) #include <bits/stdc++.h> using namespace std; const int maxn = 1e5 + 5; int a[maxn]; int L[maxn], R[maxn]; int main() { int n, k; cin >> n >> k; int ans = 0; for (int i = 1; i <= n; i++) { cin >> a[i]; ans ^= a[i]; } for (int i = 1; i <= n; i++) { L[i] = L[i - 1] ^ a[i]; } for (int i = n; i >= 1; i--) { R[i] = R[i + 1] ^ a[i]; } if (ans != 0) { cout << "YES" << endl; return 0; } for (int i = 1; i <= n; i++) { if (a[i] >= k) { int x = a[i] - k; if ((L[i - 1] ^ R[i + 1] ^ x) != 0) { cout << "YES" << endl; return 0; } } } cout << "NO" << endl; return 0; }