ZZULIOJ 2827: 欢度佳节（状态压缩）

ZZULIOJ 2827: 欢度佳节

对于17个连通块，进行爆搜，一共$2^{17}$种情况，状态压缩的思想，用01表示是否走这个连通块，然后DFS暴力搜索，判断能否完成所有的连通块

$dist[i][j]$表示点$(i,j)$是否选择
$st[i][j]$表示点$(i,j)$是否走过

#include<bits/stdc++.h>
using namespace std;

typedef pair<int,int> PII;
const int N = 6;
const int x[] = {1,1,1,1,2,2,2,3,3,3,4,4,4,5,5,5,5};
const int y[] = {1,2,4,5,2,3,4,2,3,4,2,3,4,1,2,4,5};
const int dx[] = {-1,0,1,0};
const int dy[] = {0,1,0,-1};
int g[N][N],dist[N][N],st[N][N];
int cnt,ans,sum;
vector<PII> Q;

void dfs(int sx,int sy)
{
	if(!dist[sx][sy]) return ;
	else
	{
		st[sx][sy]=1;
		for(int i=0;i<4;i++)
		{
			int x=sx+dx[i],y=sy+dy[i];
			if(x<1||x>5||y<1||y>5) continue;
			if(!st[x][y]&&dist[x][y])
			{
				sum++;
				dfs(x,y); 
			}
		}
	}
}

signed main()
{
    int T;cin>>T; 
	while(T--)
	{
		memset(g,0,sizeof g);
		for(int i=0;i<17;i++)
		{
			cin>>g[x[i]][y[i]];
			Q.push_back({x[i],y[i]});
		} 
		int n;cin>>n;
		int res=0;
		for(int i=0;i<(1<<17);i++)
		{
			memset(dist,0,sizeof dist);
			memset(st,0,sizeof st);
			cnt=ans=sum=0;
			for(int j=0;j<17;j++)//对于17种情况进行状态压缩 
				if(i>>j&1)
				{
					dist[Q[j].first][Q[j].second]=1;
					cnt++;
					ans+=(g[Q[j].first][Q[j].second]+5)/6;
				} 
			if(!dist[5][1]||ans>n) continue;//第一个连通块都不满足，或者高于筛子总次数 
			else
			{
				if(dist[5][1]) sum=1;//sum标识方案是否能够满足 
				dfs(5,1);
				if(sum==cnt)//该方案可以满足，并且总次数，不高于能用筛子的次数 
				res=max(res,cnt);
			}
		}
		cout<<res<<endl;
	} 
    return 0;
}