SSL2418 数字序列 状压DP

题目大意,求一个数列的方案数,这个数列满足,只由1到k之间的数字组成,而且当ai=aj时,i和j的差值至少为ai

之前那个50分的是因为压得方式不同

设f[i,i1,i2...ik]表示长度为i,数字1上一次出现位置在第i位的前i1位,2上次在i的前i2位……有多少种方案
当i1>=1时,i1=0。当i2>=2时,i2=0……
通过实践发现,k=7时,合法的i1,i2……ik状态最多900多种
现在来考虑转移,枚举第i位是1..k,如果第i位可以填i,就是一种转移。
通过实践发现,k=7时,合法的转移最多3000多种!
这样,我们可以预处理出f[i]到f[i+1]的每种转移,然后dp
k=7时时间复杂度o(3000*n),可以承受

显然我们需要两次dfs一次暴力转移一次暴力初始值

状态显然用7进制压缩即可

合法的状态满足没有相同数字【意外的没想到结果状态数一直是5040= =写了一天的我满脸大汗

/* ***********************************************
Author        :BPM136
Created Time  :2016-3-18 17:12:25
File Name     :E:\BPM136\BPM\DP\SSL2418GG.cpp
************************************************ */

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<set>
#include<map>
#include<bitset>
#include<string>
#include<math.h>
#include<stdlib.h>
#include<time.h>
#define LL long long
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define dfo(i,a,b) for(int i=a;i>=b;i--)
#define efo(i,x) for(int i=last[x];i!=0;i=e[i].next)
using namespace std;
LL read()
{
     LL d=0,f=1;char s=getchar();
     while(s<48||s>57){if(s==45)f=-1;s=getchar();}
     while(s>=48&&s<=57){d=d*10+s-48;s=getchar();}
     return d*f;
}
#define inf 910109
int f[2][900000];
struct S
{
	int x,y;
}s[20000];
int n,m,num=0;
int tt[10],c[10],aa[10],ma=0,mi=0;
int status[10000];
bool tot[900000];
int po7[9];

void add(int x,int y)
{
	//cout<<x<<' '<<y<<' '<<num+1<<endl;
	if(tot[x]==0)
	{
		status[++status[0]]=x;
		tot[x]=1;
	}
	s[++num].x=x;s[num].y=y;
	mi=min(mi,min(x,y));
	ma=max(ma,max(x,y));
}

void check()
{
	cout<<"CCCC:";
	fo(i,1,m-1)cout<<c[i]<<' ';
	cout<<endl;
	cout<<"AAAA:";
	fo(i,1,m)cout<<aa[i]<<' ';
	cout<<endl;
}

void dfs(int k)
{
	if(k==m+1)
	{
		int x=0,y=0;
		memset(tt,0,sizeof(tt));
		fo(i,1,m)tt[c[i]]++;
		fo(i,1,m)
		{
			if(tt[i]>1)return ;
			x+=po7[i-1]*c[i];
			if(c[i]!=0)y+=((c[i]+1)%i)*po7[i-1];
		}
		fo(i,1,m)
			if(c[i]==0)
			{
				if(i==1)add(x,y);else
				add(x,y+po7[i-1]);
			}
		return ;
	}
	fo(i,0,k-1)
	{
		c[k]=i;
		dfs(k+1);
		c[k]=0;
	}
} 

void dfs2(int k)
{
	if(k==m)
	{
		memset(aa,0,sizeof(aa));
		for(int i=m-1,j=2;i>=1;i--,j++)
		{
			if(c[i]>=j&&aa[c[i]]==0)
			{
				aa[c[i]]=j-1;
			}
		}
//		check();
		int x=0;
		fo(i,1,m)x+=po7[i-1]*aa[i];
		f[1-m%2][x]++;
//		cout<<x<<endl;
		return ;
	}
	bitset<10>a;a.reset();c[k]=1;
	dfs2(k+1);c[k]=0;
	for(int i=k-1,j=2;j<=m;i--,j++)
	{
		if(i>0)a[c[i]]=1;
		if(a[j]==0)
		{
			c[k]=j;
			dfs2(k+1);
			c[k]=0;
		}
	}
}

void prework()
{
	po7[0]=1;
	fo(i,1,8)po7[i]=po7[i-1]*7;
	memset(tot,0,sizeof(tot));
	memset(c,0,sizeof(c));
	memset(tot,0,sizeof(tot));
	dfs(1);
	memset(f,0,sizeof(f));
	memset(c,0,sizeof(c));
	dfs2(1);
}

int main()
{
    //freopen(".in","r",stdin);
    //freopen(".out","w",stdout);
    n=read(),m=read();
	prework();
	fo(i,m,n)
	{
		int t=i%2;
		fo(j,1,status[0])f[t][status[j]]=0;
		fo(j,1,num)
		{
			f[t][s[j].y]+=f[1-t][s[j].x];
			if(f[t][s[j].y]>=inf)
			{
				f[t][s[j].y]%=inf;
			}
		}
	}
	int ans=0,t=n%2;
	fo(i,1,status[0])
	{
		ans+=f[t][status[i]];
		if(ans>=inf)ans%=inf;
	}
	cout<<ans<<endl;
	return 0;
}


你可能感兴趣的:(状压dp)