题目链接:方格取数(1)
-
定义 d p [ i , j ] dp[i,j] dp[i,j] 为在前 i i i 列放置棋子,且第 i i i 列放置方式为 j j j 的最大的和
-
转移方程:
d p [ i , j ] = m a x ( d p [ i , j ] , d p [ i − 1 , k ] + s u m [ j ] ) dp[i,j]=max(dp[i,j],dp[i-1,k]+sum[j]) dp[i,j]=max(dp[i,j],dp[i−1,k]+sum[j])
(其中 k k k 和 j j j 均为有效状态,且 k k k ,, j j j 没有重叠) -
可以预处理,筛去大部分无效状态。即判断 ( i & ( i>>1) ) ==0
#include<iostream>
#include<vector>
using namespace std;
typedef long long LL;
const int N=20;
int a[N][N];
int val[1<<N],pos;
LL dp[N][1<<N];
int n;
void init()
{
pos=0;
for(int i=0;i< 1<<n;i++)
if((i & (i>>1))==0) val[pos++]=i;
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++) scanf("%d",&a[i][j]);
init();
for(int i=1;i<=n;i++)
for(int j=0;j<pos;j++)
{
LL sum=0; dp[i][val[j]]=-1e18;
for(int k=0;k<n;k++)
if(val[j] >> k & 1) sum+=a[i][k+1];
for(int k=0;k<pos;k++)
if((val[j] & val[k])==0)
dp[i][val[j]]=max(dp[i][val[j]],dp[i-1][val[k]]+sum);
}
LL ans=-1e18;
for(int i=0;i< 1<<n;i++) ans=max(ans,dp[n][i]);
printf("%lld\n",ans);
}
return 0;
}