牛客 - 膜法记录(状压dp预处理)

题目链接：点击查看

题目大意：给出一个 n * m 的矩阵，其中存在着些许敌人，现在可以进行 a 次行攻击，以及 b 次列攻击，顾名思义，每次行攻击可以消灭一整行的敌人，每次列攻击可以消灭一整列的敌人，现在问能否通过合适的搭配，使得敌人全部消灭

题目分析：因为 n 最大只有20，所以考虑状态压缩预处理，对于每一列来说，所有敌人的位置肯定都有一个状态 state ，维护一下 dp[ i ] 表示二进制下的前缀和，维护方程为 if( ( i | j ) == i ) dp[ i ] += dp[ j ] ，预处理好后就可以枚举二进制判断了，对于每个状态：

1. 需要将每个状态中的 1 进行 行攻击，所以该状态中 1 的个数需要小于等于 a
2. 剩余需要列攻击的列共有 m - dp[ i ] 个，满足该数字小于等于 b 即可

通过这个题学到了一个偷懒的函数：__builtin_popcount( i ) 以读表的形式返回32位整数 i 中有多少个 1 ，效率极高

代码：

#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
using namespace std;
      
typedef long long LL;

typedef unsigned long long ull;
      
const int inf=0x3f3f3f3f;
 
const int N=1e5+100;

char maze[25][N];

int num[(1<<20)+100];

int main()
{
#ifndef ONLINE_JUDGE
//	freopen("input.txt","r",stdin);
//	freopen("output.txt","w",stdout);
#endif
//	ios::sync_with_stdio(false);
	int w;
	cin>>w;
	while(w--)
	{
		int n,m,a,b;
		scanf("%d%d%d%d",&n,&m,&a,&b);
		for(int i=0;i<n;i++)
			scanf("%s",maze[i]);
		for(int i=0;i<1<<n;i++)
			num[i]=0;
		for(int j=0;j<m;j++)
		{
			int bit=0;
			for(int i=0;i<n;i++)
				if(maze[i][j]=='*')
					bit|=1<<i;
			num[bit]++;
		}
		for(int i=0;i<n;i++)
			for(int j=0;j<1<<n;j++)
				if(((1<<i)&j)==0)
					num[j|(1<<i)]+=num[j];
		bool flag=false;
		for(int i=0;i<1<<n;i++)
			if(__builtin_popcount(i)<=a&&m-num[i]<=b)
			{
				flag=true;
				break;
			}
		if(flag)
			puts("yes");
		else
			puts("no");
	}
	
	
	
	
	
	
	
	
	
	
	
	
	
	
	
    return 0;
}