浅谈子集和 dp SoS dp

零衣贰

已于 2023-08-07 14:32:53 修改

阅读量304

点赞数 1

分类专栏：学习笔记文章标签：算法 c++ 开发语言

于 2023-06-26 13:14:07 首次发布

本文链接：https://blog.csdn.net/weixin_73113801/article/details/131387161

版权

学习笔记专栏收录该内容

45 篇文章 1 订阅

订阅专栏

SOS dp：Sum over Subsets dp ，子集和 dp

0.定义：

子集：对于两个二进制数 $x, y$ ，

$\sube y \iff x \mid y=y$ 。

如 $1001$ 就是 $1011$ 的子集。

1.问题：

给定 $2^n$ 长度的序列 $A$ ，下标从 $0$ 到 $2^n-1$ 。
$\forall x$ ，求 $F(x)=\sum\limits_{i \sube x} A_i$ 。

2. 暴力做法

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

时间复杂度 $O(4^n)$ 。

3. 子集枚举法

枚举 $S$ 及其子集 $x$

    for(int mask=1; mask<(1<<N); mask++)
        for(int x=mask; x>0; x=(x-1)&mask)
        	F[mask] += A[x];
}

时间复杂度 $O(3^n)$ 。

$O(3^n)$ 的解释：对于所有枚举的状态，对于第 $i$ 个元素而言，有三种情况：属于 $x$ ，属于 $ma s k$ 不属于 $x$ ，不属于 $ma s k$ ，因此共 $3^n$ 种状态。

4. SoS DP 解法

我们发现在上述方法中，有一个明显的缺陷：对于一个有 $k$ 位 $0$ 的二进制数 $x$ ，它是 $2^k$ 个集合的子集，因此 $A [x]$ 会被重复计算 $2^k$ 次。我们考虑一种新的集合划分方式使得 $A [x]$ 在新的集合中尽量不被重复划分。

记 $S(mask,i)=\lbrace {x \mid x \sube mask \,\,\&\&\,\, x \oplus mask < 2^{i+1}} \rbrace$ ，也就意味着 $S (ma s k, i)$ 中的元素只会在前 $i$ 位与 $ma s k$ 不同。

例如， $S(110101,2)=\lbrace {110\textcolor{red}{100}, 110\textcolor{red}{001}, 110\textcolor{red}{101}, 110\textcolor{red}{000}} \rbrace$ 。

我们尝试去找 $S$ 集合之间的关系：

当 $ma s k$ 第 $i$ 位为 $0$ 时， $S (ma s k, i) = S (ma s k, i - 1)$ 。
当 $ma s k$ 第 $i$ 位为 $1$ 时， $\cup S(mask \oplus 2^i,i-1)$ 。

记 $dp(mask,i)=\sum\limits_{x \in S(mask,i)} {A[x]}$ ，有：

$i)=\begin{cases} dp(mask, i-1) & i^{th} \,\, bit \,\, off\\ dp(mask, i-1) + dp(mask \oplus 2^i,i-1)& i^{th} \,\, bit \,\, on\\ \end{cases}$

$d p (ma s k, - 1)$ 初始为 $A [ma s k]$ 。

对于 $F [ma s k]$ ，即为 $d p (ma s k, n - 1)$ 。

	for(int mask = 0; mask < (1<<N); mask ++)
	{
		dp[mask] = buf[mask] + 1;
		dp[mask][-1] = a[mask];
		
		for(int i=0; i<N; i++)
		{
			if(mask & (1<<i)) 
				dp[mask][i] = dp[mask][i-1] + dp[mask^(1<<i)][i-1]; 
			else 
				dp[mask][i] = dp[mask][i-1];
		}
		F[mask] = dp[mask][N];
	}

时间复杂度 $\times 2^n)$ 。

空间优化：

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

例题1：iai 互补数对

link

若 $\And b=0$ , 那么 $b$ 一定是 $a$ 按位取反的子集，SoS dp 后按补集统计即可。

#include<bits/stdc++.h> 
using namespace std;

int n, num[1000005], F[1<<20];

signed main()
{
	cin >> n;
	
	for(int i=1; i<=n; i++)
	{
		cin >> num[i];
		F[num[i]] ++ ;
	}
	
	for(int i=0; i<20; i++)
		for(int mask = 0; mask < (1<<20); mask ++)
			if(mask & (1<<i)) F[mask] += F[mask^(1<<i)]; 
	
	long long ans = 0;
	int ALL = (1<<20) - 1;
	for(int i=1; i<=n; i++)
		ans += F[ALL - num[i]];
	
	cout << ans / 2;
	return 0;
}

例题2：Compatible Numbers

link

例题 1 求数量，此题求具体的数，思路与方法已知。

#include<bits/stdc++.h> 
using namespace std;

int n, num[1000005], g[1<<22];

signed main()
{
	cin >> n;
	
	for(int mask = 0; mask < (1<<22); mask ++) g[mask] = -1;
	
	for(int i=1; i<=n; i++)
	{
		cin >> num[i];
		g[num[i]] = num[i];
	}
	
	for(int i=0; i<22; i++)
		for(int mask = 0; mask < (1<<22); mask ++)
			if(mask & (1<<i) and g[mask] == -1) g[mask] = g[mask^(1<<i)];
			
	int ALL = (1<<22) - 1;
	for(int i=1; i<=n; i++)
		cout << g[ALL-num[i]] << " ";
	return 0;
}