传送门:POJ2441
题意:将n头牛分到m个牛栏里,每头牛都有自己喜欢的牛栏,每个牛栏只能放一头牛,问共有多少种分配方案。
思路:由于n和m都很小,状压搞一搞就好了,dp[i]表示m个牛栏的状态为i的分配方案(将i转化为2进制,1表示有牛,0表示没有牛)
__builtin_popcount(k)
返回k的二进制中有多少个1
代码:
#include<stdio.h>
#include<iostream>
#include<vector>
#define ll long long
#define pb push_back
#define fi first
#define se second
#define pi acos(-1)
#define inf 0x3f3f3f3f
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define rep(i,x,n) for(int i=x;i<n;i++)
#define per(i,n,x) for(int i=n;i>=x;i--)
using namespace std;
typedef pair<int,int>P;
const int MAXN=100010;
int gcd(int a,int b){return b?gcd(b,a%b):a;}
vector<int> p[30];
int dp[1 << 21];
int main()
{
int n, m;
cin >> n >> m;
if(m < n)
{
cout << 0 << endl;
return 0;
}
int t, x;
for(int i = 0; i < n; i++)
{
scanf("%d", &t);
while(t--)
{
scanf("%d", &x);
p[i].pb(--x);
}
}
dp[0] = 1;
for(int i = 0; i < n; i++)
{
for(int j = 0; j < p[i].size(); j++)
{
t = p[i][j];
for(int k = (1 << m) - 1; k >= 0; k--)
{
if(__builtin_popcount(k) == i)//一开始忘判断这个转移条件了,一直错
{
if(k & 1 << t) continue;
dp[k | 1 << t] += dp[k];
}
}
}
}
int ans = 0;
for(int i = 0; i < (1 << m); i++)
{
if(__builtin_popcount(i) == n) ans += dp[i];
}
cout << ans << endl;
return 0;
}
题意:给出一个M*N的矩阵,元素为0表示这个地方不能种玉米,为1表示这个地方能种玉米,现在规定所种的玉米不能相邻,问一共有多少种种植方法。
思路:
和上个题差不多,将矩阵的一行压成一个状态,转移的时候枚举当前行的状态和上一行的状态进行转移,再判断一下是否满足题意就行了。
代码:
#include<stdio.h>
#include<iostream>
#define ll long long
#define pb push_back
#define fi first
#define se second
#define pi acos(-1)
#define inf 0x3f3f3f3f
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define rep(i,x,n) for(int i=x;i<n;i++)
#define per(i,n,x) for(int i=n;i>=x;i--)
using namespace std;
typedef pair<int,int>P;
const int MAXN=100010;
const int mod = 100000000;
int gcd(int a,int b){return b?gcd(b,a%b):a;}
bool is[15][15];
int dp[2][1 << 13];
int main()
{
int n, m;
cin >> n >> m;
for(int i = 0; i < n; i++)
for(int j = 0; j < m; j++)
cin >> is[i][j];
bool flag;
for(int i = 0; i < (1 << m); i++)
{
flag = 1;
for(int j = 0; j < m; j++)
{
if(is[0][j] == 0 && (i >> j) & 1) flag = 0;
if(i >> j & 1)
{
if(j < m - 1 && (i >> (j + 1) & 1)) flag = 0;
if(j > 0 &&(i >> (j - 1) & 1)) flag = 0;
}
}
dp[0][i] = flag;
}
int up = (1 << m), e = 1;
for(int i = 1; i < n; i++)
{
for(int j = 0; j < up; j++)
{
for(int k = 0; k < up; k++)
{
flag = 1;
for(int id = 0; id < m; id++)
{
if(is[i][id] == 0 && ((k >> id) & 1)) flag = 0;
if(((k >> id) & 1) && ((j >> id) & 1)) flag = 0;
if(k >> id & 1)
{
if(id < m - 1 && (k >> (id + 1) & 1)) flag = 0;
if(id > 0 &&(k >> (id - 1) & 1)) flag = 0;
}
if(flag == 0) break;
}
if(flag)
dp[e][k] += dp[e ^ 1][j], dp[e][k] %= mod;
}
}
e ^= 1;
for(int j = 0; j < up; j++)//一开始没清空,快给我wa哭了
dp[e][j] = 0;
}
e ^= 1;
int ans = 0;
for(int i = 0; i < up; i++)
ans += dp[e][i], ans %= mod;
cout << ans << endl;
return 0;
}
看大佬们博客学到了更快的方法,先预处理出所有可能的转移状态,即一行中没有两两相邻的状态,同时将每一行田地的初始状态也压成一个数字,这样比较的时候就不用循环比较了,直接&一下就好,写起来很简便,速度也快。
代码:
#include<stdio.h>
#include<iostream>
#include<string.h>
#define ll long long
#define pb push_back
#define fi first
#define se second
#define pi acos(-1)
#define inf 0x3f3f3f3f
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define rep(i,x,n) for(int i=x;i<n;i++)
#define per(i,n,x) for(int i=n;i>=x;i--)
using namespace std;
typedef pair<int,int>P;
const int MAXN=100010;
const int mod = 100000000;
int gcd(int a,int b){return b?gcd(b,a%b):a;}
int dp[2][1 << 13];
int status[1 << 13];
int field[13];
int main()
{
int n, m;
while(cin >> n >> m)
{
memset(dp, 0, sizeof(dp));
int t;
for(int i = 0; i < n; i++)
{
field[i] = 0;
for(int j = 0; j < m; j++)
{
cin >> t;
if(!t)
field[i] += 1 << j;
}
}
int cnt = 0, e = 1;
for(int i = 0; i < (1 << m); i++)
if(!(i & i << 1))//判断相邻两位是否同为1
status[cnt++] = i;
for(int i = 0; i < cnt; i++)
{
if(status[i] & field[0]) continue;
dp[0][i] = 1;
}
for(int i = 1; i < n; i++)
{
for(int j = 0; j < cnt; j++)
{
dp[e][j] = 0;
if(field[i] & status[j]) continue;
for(int k = 0; k < cnt; k++)
{
if(status[k] & field[i - 1]) continue;
if(status[j] & status[k]) continue;
dp[e][j] += dp[e ^ 1][k], dp[e][j] %= mod;
}
}
e ^= 1;
}
e ^= 1;
int ans = 0;
for(int i = 0; i < cnt; i++)
ans += dp[e][i], ans %= mod;
cout << ans << endl;
}
return 0;
}