[SWTR-03]Game

题目

传送门 to luogu

思路

感谢@Alex_Wei 的博客提供了思路（与代码）。

毫无疑问，每种质因数是独立的，只考虑某一种质因数。

把指数拿出来进行考虑，$\frac{p^x}{p^z}=p^{x-z}$ ，作除法变成了作减法。

先考虑一个简化版，对于 { a } \{a\} {a} ，每次选择一个集合，共同减去一个数，最终使其均为零。

显然，对于每一个数，它能够减去的数字就是一个 $0$-$1$ 背包。也就是说，我们已知操作了 $r$ 次（这 $r$ 次作用于哪些集合是未知的），分别减去了 $b_1,b_2,b_3,\dots ,b_r$ ，那么将 { b } \{b\} {b} 进行一次$0$-$1$背包， { a } \{a\} {a} 是结果的子集（因为对于任意一个 $a$ ，就可以从 $b$ 里面选一些，使得 $a$ 减去这些值为零）。

发现指数的大小很小（只有 ${\log }_{p}a\le {\log }_{2}a\le 20$ ），那么我们将 { a } \{a\} {a} 状压，表示 $0$-$1$背包能凑出 { a } \{a\} {a} 的最小的 { b } \{b\} {b} 。

只需要深搜即可。如果原本可以凑出的集合是 { b } \{b\} {b} ，加上 $x$ 之后，就可以凑出 { b , b + x } \{b,b+x\} {b,b+x} 。用代码来讲，就是：

s_ = s | (s << x);

然后进行两次转移：

• 如果我能凑出 { a } \{a\} {a} ，那么 ∀ { a ′ } ⊂ A \forall \{a'\}\subset A ∀{a′}⊂A ，我也可以凑出 { a ′ } \{a'\} {a′} 。
• 并没有要求全部归 $0$ 。如果我原本要凑出 { a } \{a\} {a} ，那么我也可以只凑出 { a − x } \{a-x\} {a−x} （让剩下的值都是 $x$ ）。在代码中可以只考虑 $x=1$ 的情况（类似于迭代）。

突然发现与底数无关。所以我们只用求一次即可。

代码

#include <cstdio>
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
const int base = 20, ALL = (1<<base)-1;
int f[ALL+1], state[ALL+1], cnt[ALL+1], n, ans;
void dfs(int u,int now,int S){
	if(u >= f[S]) return ; else f[S] = u;
	for(int i=now; i<=base; i++)
		dfs(u+1,i,(S|(S<<i))&ALL);
}
bool isPrime[ALL+1];
vector<int> primes;
int main(){
	/* 因为{1,2,4,8,16}能表示出所有不大于31的数
	而指数<20，所以f[i]不会超过5 */
	for(int i=0; i<=ALL; ++i) f[i] = 5;
	dfs(0,1,1<<0); // {A} = {0}时，最佳{B} = empty
	for(int s=ALL; s; --s)
		for(int i=1; i<=s; i<<=1)
			if((s&i) == i) // 转移子集
				f[s^i] = min(f[s^i],f[s]);
	for(int i=2; i<=ALL; i+=2) // 少一
		f[i] = min(f[i],f[i>>1]);
	for(int i=2; i<=ALL; ++i) isPrime[i] = true;
	for(int i=2; i<=ALL; ++i) if(isPrime[i]){
		primes.push_back(i); // 埃筛求质数
		for(int j=min(ALL+1ll,1ll*i*i); j<=ALL; j+=i)
			isPrime[j] = false;
	}
	scanf("%d",&n);
	for(int i=1,a; i<=n; i++){
		scanf("%d",&a);
		for(auto j : primes){
			if(j*j > a) break; int c = 0;
			while(a%j == 0) ++ c, a /= j;
			state[j] |= (1<<c), ++ cnt[j];
		}
		// 还有一个质因数：当前的a
		state[a] |= 1<<1, ++ cnt[a];
		// 如果当前的a是1，不会影响结果
	}
	for(auto i : primes){
		if(cnt[i] < n) // 至少有一个数的指数为0
			state[i] |= 1<<0;
		ans += f[state[i]];
	}
	printf("%d\n",ans);
	return 0;
}