问题描述
在电影《金陵十三钗》中有十二个秦淮河的女人要自我牺牲代替十二个女学生去赴日本人的死亡宴会。为了不让日本人发现,自然需要一番乔装打扮。但由于天生材质的原因,每个人和每个人之间的相似度是不同的。由于我们这是编程题,因此情况就变成了金陵n钗。给出n个女人和n个学生的相似度矩阵,求她们之间的匹配所能获得的最大相似度。
所谓相似度矩阵是一个n*n的二维数组like[i][j]。其中i,j分别为女人的编号和学生的编号,皆从0到n-1编号。like[i][j]是一个0到100的整数值,表示第i个女人和第j个学生的相似度,值越大相似度越大,比如0表示完全不相似,100表示百分之百一样。每个女人都需要找一个自己代替的女学生。
最终要使两边一一配对,形成一个匹配。请编程找到一种匹配方案,使各对女人和女学生之间的相似度之和最大。
输入格式
第一行一个正整数n表示有n个秦淮河女人和n个女学生
接下来n行给出相似度,每行n个0到100的整数,依次对应二维矩阵的n行n列。
输出格式
仅一行,一个整数,表示可获得的最大相似度。
样例输入
4
97 91 68 14
8 33 27 92
36 32 98 53
73 7 17 82
样例输出
354
数据规模和约定
对于70%的数据,n<=10
做的时候dfs一闪而过但是被我否了,觉得n<=10 这个范围太大。其实是可以的。深搜的上限大概是
dfs简直就是傻瓜,然后往dp想,然后就gg了。因为这个状态转移方程很难。难点在于每个留学生都有n个不同的秦淮河姑娘的对应方案,而且都要记录下来,那么需要记录的就有点多了。但是要保存在一个至多二维里面。这样就很尴尬了。解决办法是用二进制,
这个写得很好
对于70%的数据,直接用全排列枚举出每个女生对应的人然后求解,取最大值,时间复杂度是O(n*n!),对于n<=10,是可以接受的
对于100%的数据,可以采用状态压缩的办法进行动态规划:
dp[i][j]表示现在进行到i个女人,被占用学生的状态为j的最大制。
将状态j表示成二进制,第k位为1 表示第k个学生已经被占用了。
比如j=3,二进制是00000011,表示第1和第2个学生已经被占用。
对于初始状态:
dp[0][j]=0
转移方程是:
dp[i][j]=max(dp[i-1][k]+like[i][t])
其中t是j中其中1个1的位置,k是j把第t位变成0的数,把所有t找一边取个最大值再+like[i][t],就是dp[i][j]
答案就是dp[n][2^n-1]
时间复杂度应该是O(n*2^n)
考虑到对于所有i,只和i-1项有关,所以可以减少一维空间消耗
对于n<=13是可以秒出的。
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
#define MAXN 15
const int maxzt=(1<<13);
int dp[maxzt];
int n;
int like[MAXN][MAXN];
int numberOfOne(int num)
{
int cnt=0;
while(num){
cnt+=(num&1);
num>>=1;
}
return cnt;
}
int lowbit(int x) {return (x)&(-x);}
int posOfOne(int num)
{
int pos=0;
while(num)
{
pos++;
if(num&1)
return pos;
num>>=1;
}
return pos;
}
void work(int x)
{
int maxstatus=1<<n;
for(int i=0;i<maxstatus;i++)
{
int nowstatus=i,t=numberOfOne(nowstatus);
if(t!=x) continue;
while(t--)
{
int pos=lowbit(nowstatus);
dp[i]=max(dp[i],dp[i-pos]+like[x][posOfOne(pos)]);
nowstatus-=pos;
}
}
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
cin>>like[i][j];
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
work(i);
printf("%d\n",dp[(1<<n)-1] );
}
状压dp代码
#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define ff(i,a,b) for(int i = a; i <= b; i++)
#define f(i,a,b) for(int i = a; i < b; i++)
typedef pair<int,int> P;
#define ll long long
int like[22][22];
int dp[2][(1<<20) +10];
int main()
{
ios::sync_with_stdio(false);
int n;
cin >> n;
ff(i,1,n) ff(j,1,n)
cin >>like[i][j];
int tt = (1 << n) - 1;
ff(i,1,n) ff(j,1,tt) f(k,0,n)
if(j & (1<<k)) dp[i][j] = max(dp[i][j],dp[i - 1][j - (1<<k)] + like[i][k + 1]);
cout << dp[n][tt] << endl;
return 0;
}