【記憶化搜索】PKU-2044-Weather Forecast

經過這題，我發現原來用vector做參數是會超時的，難怪沒人這樣用來啦……

題目

#include<iostream>
#include<vector>
using namespace std;
struct T
{
	int a[4];
};
int n,p[370][20],b[5]={1,4,13,16};
vector<int> v1[15],v2[15];
bool hash[10][370][7][7][7][7];
int dfs(int k,int d,T s)
{
	if(d>n)return 1;
	int i,j,vis[20];
	bool &ret=hash[k][d][s.a[0]][s.a[1]][s.a[2]][s.a[3]];
	T ss;
	if(ret)return 0;
	ret=true;
	for(i=0;i<v1[k].size();i++)
		if(p[d][v1[k][i]])return 0;
	for(i=0;i<4;i++)
		if(s.a[i]==7)return 0;
	for(i=0;i<v2[k].size();i++)
	{
		memset(vis,0,sizeof(vis));
		for(j=0;j<v1[v2[k][i]].size();j++)vis[v1[v2[k][i]][j]]=1;
		ss=s;
		for(j=0;j<4;j++)
		{
			if(vis[b[j]])ss.a[j]=0;
			else ss.a[j]++;
		}
		if(dfs(v2[k][i],d+1,ss))return 1;
	}
	return 0;
}
int main()
{
	//freopen("a.txt","r",stdin);
	//freopen("b.txt","w",stdout);
	int i,j,l;
	T s;
	for(i=1;i<10;i++)
	{
		v1[i].clear();
		l=i;
		l+=(i-1)/3;
		v1[i].push_back(l);
		v1[i].push_back(l+1);
		v1[i].push_back(l+4);
		v1[i].push_back(l+5);
	}
	for(i=1;i<10;i++)
	{
		v2[i].clear();
		v2[i].push_back(i);
		if(i-1>0&&i%3!=1)v2[i].push_back(i-1);
		if(i+1<10&&i%3)v2[i].push_back(i+1);
		if(i-3>0)v2[i].push_back(i-3);
		if(i+3<10)v2[i].push_back(i+3);
		if(i-2>0&&i%3==0)v2[i].push_back(i-2);
		if(i+2<10&&i%3==1)v2[i].push_back(i+2);
		if(i-6>0)v2[i].push_back(i-6);
		if(i+6<10)v2[i].push_back(i+6);
	}
	while(scanf("%d",&n)&&n)
	{
		for(i=1;i<=n;i++)
			for(j=1;j<=16;j++)
				scanf("%d",&p[i][j]);
		
		for(i=0;i<4;i++)s.a[i]=1;
		memset(hash,false,sizeof(hash));	
		printf("%d\n",dfs(5,1,s));
	}
	return 0;
}