2020ICPC 江西省赛 L.WZB‘s Harem（状压dp）

题目传送门

题意： 给你一个$n\ast n(n\le 20)$的图，图上有$0$和$1$两种数字，现在有$n$个互不相同的皇后，你需要将这$n$个皇后放在$0$的位置上，并且保证这$n$个皇后之间不存在任意两个皇后处于同一行、同一列，问你有多少种放法。

思路： 很像n皇后问题，但是数据太大了爆搜肯定不行。但是这个数据很容易让人想到状压dp。我们可以考虑用$f[i][j]$表示放完第$i$行，现在状态为$j$的方案数。状态$j$可以描述为前$i$行放置的皇后的位置的二进制状压。

问题来了，我们需不需要考虑将原来地图上是1的位置加入到状态中？答案是不需要的，因为我们按行遍历，每一行放1个皇后，那么当前行中原来就是1的那个只会影响这一行的皇后不能放在该位置，对其他不产生影响。

我们按照行数枚举，再枚举当前放的位置，接着枚举上一层的状态，如果合法就转移。这里需要注意的是，我们枚举上一层的状态的时候不需要将所有状态都枚举，比如我们当前枚举到了第$i$行，那么上一行的状态的二进制中一定只有$i-1$个1，我们只需要先对所有状态进行预处理，对二进制中1的个数进行分类，枚举状态的时候就可以省去很多无效状态。

因为第$i$层的状态只会由第$i-1$层转移过来，所以我们还可以将dp数组的第一维省去，节省空间。

代码：

#define int long long
const int N = 1e6+1e5;
const int inf = 0x3f3f3f3f;
const int mod = 1e9+7;

int a[22];
int f[22][N];
vector<int>v[22];
signed main()
{
	int n;
	cin>>n;
	for(int i=0;i<(1<<n);i++)
	{
		int cur = 0,m = i;
		while(m)
		{
			if(m&1)
				cur++;
			m>>=1;
		}
		v[cur].pb(i);
	}
	for(int i=0; i<n; i++)
	{
		for(int j=0;j<n;j++)
		{
			int x;
			cin>>x;
			a[i] |= (x<<j);
		}
	}
	for(int i=0;i<n;i++)
	{
		if((1<<i)&a[0]) continue;
		f[0][1<<i] = 1;
	}
	for(int i=1;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			if((1<<j)&a[i]) continue;
			for(auto k:v[i])
			{
				if((1<<j)&k) continue;
				f[i][k|(1<<j)] += f[i-1][k];
				f[i][k|(1<<j)] %= mod;
				//cout<<i<<" "<<(k|(1<<j))<<" "<<f[i][k|(1<<j)]<<endl;
			}
		}
	}
	int tot = 0;
	for(auto i:v[n])
		tot = (tot + f[n-1][i])%mod;
	for(int i=1;i<=n;i++)
	 	tot = (tot * i) % mod;
	cout<<tot<<endl;


	return 0;
}