一分钟学会SOS DP,学不会来打我

SOS DP

又叫子集$dp$

有时候会遇到这样的问题,已知数组$a$,对每一个$mask$求

$f[mask]=\sum _{x\&mask=x}{a}_x$

你当然可以暴力枚举$mask$和$x$求,但是这个玩意其实可以$dp$

有点类似$FWT$

定义$f[mask][i]$表示只有二进制前$i$位和$mask$不同的满足条件的$\sum a_x$

那么当$mask$的第$i$位为$0$时

满足条件的$x$第$i$位也必然为$0$,那么实际上就等价于$f[mask][i-1]$

因为如果$x$第$i$位和$mask$相等,就相当于只有前$i-1$位不同而已

当$mask$的第$i$位为$1$时

那么满足条件的$x$第$i$位可以为$0$也可以为$1$

如果$x$第$i$位为$1$,那么实际上等价于$f[mask][i-1]$(同上)

如果$x$第$i$位为$0$,我们仍然构造一种方案使得能够类似上面来转移

也就是等价于$f[mask\oplus 2^i][i-1]$

$解释$

现在我们要找的$x$满足$x\&mask=x$

现在我们暂时不考虑第$i$位二进制是否满足

那么$f[mask\oplus 2^i][i-1]$中的所有$x$,不考虑第$i$位时,仍然满足$x\&mask=x$

现在考虑第$i$位,因为$mask\oplus 2^i$的第$i$位是$0$

那么$f[mask\oplus 2^i]$中满足条件的$x$的第$i$位都是$0$

显然这样的$x$就是我们所需要的$x$

所以有转移方程$f[mask][i]=f[mask][i-1]+f[mask\oplus 2^i][i-1]$

可以滚动数组优化空间,枚举$i$,再枚举$mask$即可

因为每次从$i,mask$更小的地方转移而来,所以$mask$应该倒序枚举保证是上一轮的状态

for(int i = 0; i<(1<<N); ++i)	F[i] = A[i];

for(int i = 0;i < N; ++i)
for (int mask = (1<<N)-1; mask >= 0 ; mask--)
	if (mask & (1 << i))
		F[mask] += F[mask ^ (1 << i)];

但是其实$mask$正序枚举也是 可以的…因为$mask\oplus 2^i$一定是直接继承的上一次的状态,所以不会有影响

for(int mask = 0; mask < (1<<N); ++mask){
	if(mask & (1<<i))	F[mask] += F[mask^(1<<i)];

事实上网上博客基本都是正序来写的…虽然这样没问题,但是不利于理解吧??..

---

例题

没有链接了,没有评测点

题意

给定$n$个数,保证$n$和数字都小于等于$10^6$

求$a_i\&a_j==0$的$(i,j)$的数目.$i,j$无序,贡献算两次

我们把每个数字$x$的数目叠起来作为$a_x$,然后求

$f[mask]=\sum _{x\&mask=x}{a}_x$就可以了

但是$f[mask]$表示那些是$mask$子集的$a_x$的和

现在我们是求不是子集的和,那么也就是求$mask$的补集的$f$值即可

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1<<20;
int t,n,a[maxn<<1],f[maxn<<1],mx=(1<<20);
int main()
{
	cin >> t;
	while( t-- )
	{
		cin >> n;
		for(int i=1;i<=n;i++)
		{
			int x; scanf("%d",&x);
			a[x]++;
		}
		for(int i=0;i<mx;i++)	f[i] = a[i];
		for(int i=0;i<20;i++)
		{
			for(int j=mx-1;j>=0;j--)
				if( j&(1<<i) )	f[j] += f[j^(1<<i)];
		}
		long long ans = 0;
		for(int i=0;i<=1000000;i++)	ans += 1ll*a[i]*f[(mx-1)^i],a[i]=0;
		cout << ans << endl;
	}
}