问题
将n个整数分成若干组,并使得各组的数字之和为质数,求分组的方案数。
- 1 ≤ n ≤ 15 1\leq n \leq 15 1≤n≤15
- 整数 a i a_i ai 满足 1 ≤ a i ≤ 99 1\leq a_i \leq 99 1≤ai≤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 ) O(2^4) O(24)
复杂度
- 集合的可能数是 2 n 2^n 2n,其中,含有 k k k 个元素的集合数是 C n k C_n^k Cnk
- 对于含有 k k k 个元素的集合的划分,复杂度 O ( 2 k − 1 ) O(2^{k-1}) O(2k−1)
- 总体复杂度: O ( ∑ k = 1 n C n k ⋅ 2 k − 1 ) = O ( 3 n ) O(\sum_{k=1}^nC_n^k\cdot 2^{k-1})=O(3^n) O(∑k=1nCnk⋅2k−1)=O(3n)
代码
/* 集合划分 状压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;
}