集合划分（状压DP）

问题

将n个整数分成若干组，并使得各组的数字之和为质数，求分组的方案数。

• $1\le n\le 15$
• 整数 $a_i$ 满足 $1\le a_i\le 99$

分析

集合划分

• 划分 { a 1 ， a 2 ， a 3 ， a 4 ， a 5 } \{a_1，a_2，a_3，a_4，a_5\} {a1​，a2​，a3​，a4​，a5​}
  • { a 1 } \{a_1\} {a1​} 和 划分 { a 2 ， a 3 ， a 4 ， a 5 } \{a_2，a_3，a_4，a_5\} {a2​，a3​，a4​，a5​}
  • { a 1 、 a 2 } \{a_1、a_2\} {a1​、a2​} 和 划分 { a 3 ， a 4 ， a 5 } \{a_3，a_4，a_5\} {a3​，a4​，a5​}
  • { a 1 、 a 3 } \{a_1、a_3\} {a1​、a3​} 和 划分 { a 2 ， a 4 ， a 5 } \{a_2，a_4，a_5\} {a2​，a4​，a5​}
  • { a 1 、 a 4 } \{a_1、a_4\} {a1​、a4​} 和 划分 { a 2 ， a 3 ， a 5 } \{a_2，a_3，a_5\} {a2​，a3​，a5​}
  • { a 1 、 a 5 } \{a_1、a_5\} {a1​、a5​} 和 划分 { a 2 ， a 3 ， a 4 } \{a_2，a_3，a_4\} {a2​，a3​，a4​}
  • { a 1 、 a 2 、 a 3 } \{a_1、a_2、a_3\} {a1​、a2​、a3​} 和 划分 { a 4 ， a 5 } \{a_4，a_5\} {a4​，a5​}
  • $\cdots$
• 若集合 { a 1 ， a 2 ， a 3 ， a 4 ， a 5 } \{a_1，a_2，a_3，a_4，a_5\} {a1​，a2​，a3​，a4​，a5​} 的子集划分已经知道，则此集合的划分复杂度为$O(2^4)$

复杂度

• 集合的可能数是 $2^n$，其中，含有 $k$ 个元素的集合数是 $C_n^k$
• 对于含有 $k$ 个元素的集合的划分，复杂度 $O(2^{k-1})$
• 总体复杂度：$O({\sum }_{k=1}^n{C}_n^k\cdot 2^{k-1})=O(3^n)$

代码

/* 集合划分 状压dp */
#include<bits/stdc++.h>
using namespace std;
const int MXS = (1<<15)+5;
const int MXN = 17;
int N, a[MXN], dp[MXS], sum[MXS]={0};
int c[MXS], p[1700] = {1, 1}; // 高位1的位置、是否质数
int main(){
	int t, T = 0, S;
	for(int i = 0; i < 15; ++i) c[1<<i] = i+1;
	for(int i = 2; i <= 850; ++i) // 埃氏筛法
		for(int j = i<<1; j < 1700; j += i) p[j] = 1;
	scanf("%d", &t);	
	while(t--){
		scanf("%d", &N);			
		for(int i = 1; i <= N; ++i) scanf("%d", a+i);
		S = (1<<N)-1;
		for(int s = 1; s <= S; ++s){ // 枚举集合
			int ls = -s&s, hs = s^ls;
			if(ls == s) sum[s] = a[c[s]]; // 单个数字的组合
			else sum[s] = sum[hs] + sum[ls]; // 多个数字的组合
			dp[s] = p[sum[s]] == 0; // 集合不拆分是否质数
			for(int ss = hs; ss > 0; ss = (ss-1)&hs) // 枚举组合状态
				dp[s] += dp[ss]*(p[sum[ss^s]]==0);
		}
		printf("%d\n", dp[S]);
	}
    return 0;
}