SCOI2005互不侵犯King 状压Dp

一脸懵逼,爆int,然而在数据范围内的我出的几组测试数据都是负数= =

好吧不多说,用邻接表优化转移,还有就是比较的时候si不但直接和sj比,还直接和sj/2和sj*2比就行

复杂度倒是很正常丝毫不会爆

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<algorithm>
#include<bitset>
#define LL long long
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define down(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<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
	while(s>='0'&&s<='9'){d=d*10+s-'0';s=getchar();}
	return d*f;
}
#define N 10
#define M 90
#define K 100
struct edge
{
	int y,next;
}e[K*K];
int last[K],ne=0;
unsigned long s[M],nu=0,d[M];
LL f[N][K][M];
int n,m;

bitset<N>t;
void dfs(int de)
{
	if(de>n)
	{
		s[++nu]=t.to_ulong();
		d[nu]=t.count();
		return;
	}
	t[de-1]=0;
	dfs(de+1);
	t[de-1]=1;
	dfs(de+2);
	t[de-1]=0;
}

void add(int x,int y)
{
	e[++ne].y=y;e[ne].next=last[x];last[x]=ne;
}

void prework()
{
	fo(i,1,nu)
	fo(j,1,nu)
	if(!(s[i]&s[j]))
	if(!(s[i]&(s[j]>>1)))
	if(!(s[i]&(s[j]<<1)))
	add(i,j);
}

void DP()
{
	fo(i,1,nu)f[1][d[i]][i]=1;
	fo(i,2,n)
	fo(cnt,0,m)
	fo(j,1,nu)
	{
		efo(k,j)
		{
			int p=e[k].y;
			if(d[j]+d[p]<=m)
			f[i][cnt+d[j]][j]+=f[i-1][cnt][p];
		}
	}
}

void writ()
{
	LL ans=0;
	fo(i,1,nu)ans+=f[n][m][i];
	printf("%lld\n",ans);
}

int main()
{
	scanf("%d%d",&n,&m);
	if(m==0)
	t.reset();
	dfs(1);
	prework();
	DP();
	writ();
	return 0;
}


你可能感兴趣的:(SCOI2005互不侵犯King 状压Dp)