本文涉及知识点
C++贪心
位运算、状态压缩、枚举子集汇总 线性基
P4301 [CQOI2013] 新Nim游戏
题目描述
传统的 Nim 游戏是这样的:有一些火柴堆,每堆都有若干根火柴(不同堆的火柴数量可以不同)。两个游戏者轮流操作,每次可以选一个火柴堆拿走若干根火柴。可以只拿一根,也可以拿走整堆火柴,但不能同时从超过一堆火柴中拿。拿走最后一根火柴的游戏者胜利。
本题的游戏稍微有些不同:在第一个回合中,双方可以直接拿走若干个整堆的火柴。可以一堆都不拿,但不可以全部拿走。从第二个回合(又轮到第一个游戏者)开始,规则和 Nim 游戏一样。
如果你先拿,怎样才能保证获胜?如果可以获胜的话,还要让第一回合拿的火柴总数尽量小。
输入格式
第一行为整数 kkk,即火柴堆数。
第二行包含 kkk 个整数 aia_iai,即各堆的火柴个数。
输出格式
输出第一回合拿的火柴数目的最小值。如果不能保证取胜,输出 −1-1−1。
输入输出样例 #1
输入 #1
6
5 5 6 6 5 5
输出 #1
21
说明/提示
数据规模与约定
对于全部的测试点,保证 1≤k≤1001 \leq k \leq 1001≤k≤100,1≤ai≤1091 \leq a_i \leq 10^91≤ai≤109。
线性基
原版游戏的胜负
性质一:如果操作前,所有数异或和为0,且至少有一堆石子,则一定能让异或和变成非0,且只能变成非0。
性质二;如果操作前,所有数异或和为x,则一定能让异或和变成0。x≠0x \neq 0x=0令x的最高位为i,则x=2i+y,0≤y<2ix= 2^i+y,0\le y < 2^ix=2i+y,0≤y<2i一定存在数has==2i+z,z≥0==2^i+z,z\ge 0==2i+z,z≥0,令新值为u,则异或和为y⊕z⊕u==0,即u=y⊕zy\oplus z \oplus u==0,即u = y\oplus zy⊕z⊕u==0,即u=y⊕z
y⊕z≤y+z<2i+zy\oplus z\le y+z < 2^i+zy⊕z≤y+z<2i+z,即u<hasu < hasu<has,has减少has-u。
性质一:如果当前数字异或为0,则一定不是一堆,当前回合当前方赢不了。
结论一:初始异或和为0,先手方必败。先手方操作操作后异或和必读非0,后手方最优解变成非0。数字不断变小,一定在后手方那变成0.
结论二:如果初始异或和不为0,按性质二操作,后手方必败。
原理
“你”必胜,留下任意一堆。如何让“你”第一回合拿的尽快能的少?
如果若干数字能异或成0,必须取取走一个,否则后手方只保留这几个。
这若干个数,取走那个都后继状态的影响都一样,故贪心的取走最小的数。
实现
线性基记录所有保留的石头。
将数按降序排序,通过x枚举,如果线性基的结果包括x,必须取走。否则保留。
代码
核心代码
#include <iostream>
#include <sstream>
#include <vector>
#include<map>
#include<unordered_map>
#include<set>
#include<unordered_set>
#include<string>
#include<algorithm>
#include<functional>
#include<queue>
#include <stack>
#include<iomanip>
#include<numeric>
#include <math.h>
#include <climits>
#include<assert.h>
#include<cstring>
#include<list>#include <bitset>
using namespace std;template<class T1, class T2>
std::istream& operator >> (std::istream& in, pair<T1, T2>& pr) {in >> pr.first >> pr.second;return in;
}template<class T1, class T2, class T3 >
std::istream& operator >> (std::istream& in, tuple<T1, T2, T3>& t) {in >> get<0>(t) >> get<1>(t) >> get<2>(t);return in;
}template<class T1, class T2, class T3, class T4 >
std::istream& operator >> (std::istream& in, tuple<T1, T2, T3, T4>& t) {in >> get<0>(t) >> get<1>(t) >> get<2>(t) >> get<3>(t);return in;
}template<class T = int>
vector<T> Read() {int n;cin >> n;vector<T> ret(n);for (int i = 0; i < n; i++) {cin >> ret[i];}return ret;
}
template<class T = int>
vector<T> ReadNotNum() {vector<T> ret;T tmp;while (cin >> tmp) {ret.emplace_back(tmp);if ('\n' == cin.get()) { break; }}return ret;
}template<class T = int>
vector<T> Read(int n) {vector<T> ret(n);for (int i = 0; i < n; i++) {cin >> ret[i];}return ret;
}template<int N = 1'000'000>
class COutBuff
{
public:COutBuff() {m_p = puffer;}template<class T>void write(T x) {int num[28], sp = 0;if (x < 0)*m_p++ = '-', x = -x;if (!x)*m_p++ = 48;while (x)num[++sp] = x % 10, x /= 10;while (sp)*m_p++ = num[sp--] + 48;AuotToFile();}void writestr(const char* sz) {strcpy(m_p, sz);m_p += strlen(sz);AuotToFile();}inline void write(char ch){*m_p++ = ch;AuotToFile();}inline void ToFile() {fwrite(puffer, 1, m_p - puffer, stdout);m_p = puffer;}~COutBuff() {ToFile();}
private:inline void AuotToFile() {if (m_p - puffer > N - 100) {ToFile();}}char puffer[N], * m_p;
};template<int N = 1'000'000>
class CInBuff
{
public:inline CInBuff() {}inline CInBuff<N>& operator>>(char& ch) {FileToBuf();ch = *S++;return *this;}inline CInBuff<N>& operator>>(int& val) {FileToBuf();int x(0), f(0);while (!isdigit(*S))f |= (*S++ == '-');while (isdigit(*S))x = (x << 1) + (x << 3) + (*S++ ^ 48);val = f ? -x : x; S++;//忽略空格换行 return *this;}inline CInBuff& operator>>(long long& val) {FileToBuf();long long x(0); int f(0);while (!isdigit(*S))f |= (*S++ == '-');while (isdigit(*S))x = (x << 1) + (x << 3) + (*S++ ^ 48);val = f ? -x : x; S++;//忽略空格换行return *this;}template<class T1, class T2>inline CInBuff& operator>>(pair<T1, T2>& val) {*this >> val.first >> val.second;return *this;}template<class T1, class T2, class T3>inline CInBuff& operator>>(tuple<T1, T2, T3>& val) {*this >> get<0>(val) >> get<1>(val) >> get<2>(val);return *this;}template<class T1, class T2, class T3, class T4>inline CInBuff& operator>>(tuple<T1, T2, T3, T4>& val) {*this >> get<0>(val) >> get<1>(val) >> get<2>(val) >> get<3>(val);return *this;}template<class T = int>inline CInBuff& operator>>(vector<T>& val) {int n;*this >> n;val.resize(n);for (int i = 0; i < n; i++) {*this >> val[i];}return *this;}template<class T = int>vector<T> Read(int n) {vector<T> ret(n);for (int i = 0; i < n; i++) {*this >> ret[i];}return ret;}template<class T = int>vector<T> Read() {vector<T> ret;*this >> ret;return ret;}
private:inline void FileToBuf() {const int canRead = m_iWritePos - (S - buffer);if (canRead >= 100) { return; }if (m_bFinish) { return; }for (int i = 0; i < canRead; i++){buffer[i] = S[i];//memcpy出错 }m_iWritePos = canRead;buffer[m_iWritePos] = 0;S = buffer;int readCnt = fread(buffer + m_iWritePos, 1, N - m_iWritePos, stdin);if (readCnt <= 0) { m_bFinish = true; return; }m_iWritePos += readCnt;buffer[m_iWritePos] = 0;S = buffer;}int m_iWritePos = 0; bool m_bFinish = false;char buffer[N + 10], * S = buffer;
};template<class T = long long, int BITCNT = 63 >
class CLinearBasis {
public:CLinearBasis() :m_bit(BITCNT, -1) { }void Addd(T x) {for (int i = BITCNT - 1; i >= 0; i--) {if (((T)1 << i) & x) {if (-1 == m_bit[i]) {m_bit[i] = x;return;}else {x ^= m_bit[i];}}}};T Max() {T ret = 0;for (int i = BITCNT - 1; i >= 0; i--) {if (-1 == m_bit[i]) { continue; }if (((T)1 << i) & ret) { continue; }ret ^= m_bit[i];}return ret;}bool Can(T x) const {for (int i = BITCNT - 1; i >= 0; i--) {if (((T)1 << i) & x) {if (-1 == m_bit[i]) { return false; }x ^= m_bit[i];}}return true;};
protected:vector<T> m_bit;
};class Solution {public:long long Ans(vector<int>& nums) {sort(nums.begin(), nums.end(),greater<>());CLinearBasis<> lb;long ans = 0;for (const auto& i:nums) {if (lb.Can(i)) { ans += i; continue; }lb.Addd(i); }return ans;};};int main() {
#ifdef _DEBUGfreopen("a.in", "r", stdin);
#endif // DEBUGios::sync_with_stdio(0); cin.tie(nullptr); cout.tie(nullptr); auto nums = Read<int>();
#ifdef _DEBUG//printf("N=%d", N);Out(nums, ",nums=");//Out(B, ",B=");//Out(ope, ",ope=");
#endif // DEBUGauto res = Solution().Ans(nums);cout << res << "\n";return 0;
}
单元测试
vector<pair<long long, int>> im;TEST_METHOD(TestMethod1){vector<int> nums = { 5,5,6,6,5,5 };auto res = Solution().Ans(nums);AssertEx(21LL, res);}
扩展阅读
我想对大家说的话 |
---|
工作中遇到的问题,可以按类别查阅鄙人的算法文章,请点击《算法与数据汇总》。 |
学习算法:按章节学习《喜缺全书算法册》,大量的题目和测试用例,打包下载。重视操作 |
有效学习:明确的目标 及时的反馈 拉伸区(难度合适) 专注 |
员工说:技术至上,老板不信;投资人的代表说:技术至上,老板会信。 |
闻缺陷则喜(喜缺)是一个美好的愿望,早发现问题,早修改问题,给老板节约钱。 |
子墨子言之:事无终始,无务多业。也就是我们常说的专业的人做专业的事。 |
如果程序是一条龙,那算法就是他的是睛 |
失败+反思=成功 成功+反思=成功 |
视频课程
先学简单的课程,请移步CSDN学院,听白银讲师(也就是鄙人)的讲解。
https://edu.csdn.net/course/detail/38771
如何你想快速形成战斗了,为老板分忧,请学习C#入职培训、C++入职培训等课程
https://edu.csdn.net/lecturer/6176
测试环境
操作系统:win7 开发环境: VS2019 C++17
或者 操作系统:win10 开发环境: VS2022 C++17
如无特殊说明,本算法用**C++**实现。