poj 3740 Easy Finding 搜索+位运算

题目大意:有一个01矩阵,问是否可以从矩阵中选出某些行,使得选出的这些行所构成的矩阵的每一列都只有一个1。

初看题目,没有好的办法,搜索吧(暴力的),由于题目中给出M ≤ 16, N ≤ 300,所以可以枚举行的所有选择情况,当然在这里就不是深搜去产生所有的选择情况了,直接通过整数来表示就行了(1对应于选择第一行,2对应于选择第二行,3对应于选择第一行和第二行。。。。。),那么要怎么样来判断所选择出来的那些行构成的矩阵的那些列是否只含有一个1呢?这里位运算就显示出它的强大之处了,首先在读入的时候,就把矩阵压缩成一个N个数的数组(直观的理解就是把初始矩阵的每一列看成一个十进制数的二进制表示,现在把它还原成十进制),然后在搜索的过程中,把数列里面的每个数和当前的枚举整数(表示选择的行)进行按位与,如果产生的整数的二进制表示中只含有一个1,那么表示选择出来的这个矩阵的这一列是可以的,如果每一列都可以,那么需要的矩阵找到了,当然在判断一个数v的二进制表示中是否只有一个1也可以有点技巧,那就是直接和v-1去按位与,如果为0,表示只有一个1,否则不是(想想为什么。。。。)。

#include <cstdio>
#include <cstring>

int col[500], M, N;

int main()
{
	int val;
	while(scanf("%d %d", &M, &N) == 2)
	{
		memset(col, 0, sizeof(col));
		bool flag = true;
		for(int i=0; i<M; i++)
		{
			for(int j=1; j<=N; j++)
			{
				scanf("%d", &val);
				col[j] = col[j]*2 + val;
				if(i == M-1 && col[j] == 0)
					flag = false;
			}
		}
		if(!flag)	{printf("It is impossible\n");continue;}//一点点剪枝,如果col[j]为0,随便怎么选都不会有1
		flag = false;
		for(int i=1; i<(1<<M); i++)
		{
			bool tmp = true;
			for(int j=1; j<=N; j++)
			{
				int t = col[j] & i;
				if(t == 0 || (t & (t-1)) != 0){tmp = false; break;}
			}
			if(tmp)	{flag = true;break;}
		}
		printf("%s\n", flag ? "Yes, I found it" : "It is impossible");
	}
	return 0;
}


你可能感兴趣的:(poj 3740 Easy Finding 搜索+位运算)