腾讯推出了一款益智类游戏——消消乐。游戏一开始,给定一个长度为 n 的序列,其中第 i 个数为 Ai。
游戏的目标是把这些数全都删去,每次删除的操作为:选取一段连续的区间,不妨记为 [L,R],如果这一段区间内所有数的最大公约数 ≥k(k 值在游戏的一开始会给定),那么这一段区间就能被直接删去。
注意:一次删除以后,剩下的数会合并成为一个连续区间。
定义 f(i) 为进行 i 次操作将整个序列删完的方案数。
你需要实现一个程序,计算 ∑i=1n(f(i)∗i) mod 1000000007。
输入格式
第一行输入两个整数 n,k(1≤n≤18)。
第二行输入 n 个正整数 ai(1≤ai≤105),表示初始序列中的每个数。
输入数据保证 1≤k≤min(a1,a2,…an)。
输出格式
输出一个整数,表示算出的答案。
样例说明
对于样例 1 而言,f(1)=1,f(2)=9,f(3)=26,f(4)=24。
对于样例 2,f(1)=0,f(2)=2。
样例输入1
4 1 1 1 1 1
样例输出1
193
样例输入2
2 2 2 3
样例输出2
4
样例输入3
1 233 233
样例输出3
1
思路:
看到数据范围不大,我们肯定要做状压dp的。
设定dp【i】【j】表示进行了i次操作,我们状态为j的方案数。
那么不难推出其状态转移方程:dp【i+1】【q】+=dp【i】【j】;
那么我们O(n^2)枚举下一步要删除的字符串起始和终点位子,那么对应我们只要维护一下其中Gcd的值即可。
并且注意,不要加了重复的部分,以及不要重复的转移。
最后注意,模后*i是要爆int的,所以我们改成加i次即可。
Ac代码:
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
int a[50];
int dp[19][262144+5000];
int mod=1000000007;
int gcd(int a,int b){
return __gcd(a,b);
}
int main()
{
int n,k;
while(~scanf("%d%d",&n,&k))
{
int end=(1<<n);
for(int i=0;i<n;i++)scanf("%d",&a[i]);
memset(dp,0,sizeof(dp));
dp[0][0]=1;
for(int i=0;i<n;i++)
{
for(int j=0;j<end;j++)
{
if(dp[i][j]>0)
{
for(int l=0;l<n;l++)
{
int val=0;
int q=j;
int flag=0;
if((j&(1<<l))>0)continue;
for(int r=l;r<n;r++)
{
if(flag==0)
{
if((j&(1<<r))==0)
{
q+=(1<<r);
val=gcd(val,a[r]);
if(val>=k)
{
dp[i+1][q]+=dp[i][j];
dp[i+1][q]%=mod;
}
}
else
{
flag=1;
}
}
else if(flag==1)
{
if((j&(1<<r))==0)
{
q+=(1<<r);
val=gcd(val,a[r]);
if(val>=k)
{
dp[i+1][q]+=dp[i][j];
dp[i+1][q]%=mod;
}
}
else
{
flag=0;
}
}
}
}
}
}
}
int output=0;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=i;j++)
{
output+=dp[i][end-1];
output%=mod;
}
}
printf("%d\n",output);
}
}