NYOJ832 合并游戏 状压DP+记忆化搜索

合并游戏

时间限制： 1000 ms  |  内存限制： 65535 KB

难度： 4

描述

    大家都知道Yougth除了热爱编程之外，他还有一个爱好就是喜欢玩。

    某天在河边玩耍的时候，他发现了一种神奇的石子，当把两个石子放在一起的时候，后一个石子会消失，而且会蹦出一定数量的金币，这可乐坏了Yougth，但是他想得到最多的金币，他该怎么做?

输入

首先一行，一个n（1<=n<=10），表示有n个石子。
接下来n*n的一个矩阵，Aij表示第i个和第j个合并蹦出的金币值(小于10000，注意合并后j会消失)。

输出

输出最多能得到的金币值。

样例输入

2
0 4
1 0
3
0 20 1
12 0 1
1 10 0

样例输出

4
22

来源

Yougth原创

上传者

TC_杨闯亮

思路：数据较小，想到状态压缩，110010中的0表示该位置的石子已经合并了。

代码：

#include <iostream>
#include <functional>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <string>
#include <map>
#include <stack>
#include <vector>
#include <set>
#include <queue>
#pragma comment (linker,"/STACK:102400000,102400000")
#define pi acos(-1.0)
#define eps 1e-6
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
#define FRE(i,a,b)  for(i = a; i <= b; i++)
#define FREE(i,a,b) for(i = a; i >= b; i--)
#define FRL(i,a,b)  for(i = a; i < b; i++)
#define FRLL(i,a,b) for(i = a; i > b; i--)
#define mem(t, v)   memset ((t) , v, sizeof(t))
#define sf(n)       scanf("%d", &n)
#define sff(a,b)    scanf("%d %d", &a, &b)
#define sfff(a,b,c) scanf("%d %d %d", &a, &b, &c)
#define pf          printf
#define DBG         pf("Hi\n")
typedef long long ll;
using namespace std;

#define INF 0x3f3f3f3f
#define mod 1000000009
const int maxn = 1050;
const int MAXN = 2005;
const int MAXM = 200010;
const int N = 1005;

int dp[maxn],a[12][12];
int n;

int isok(int pre,int now) //判断下一状态是否合法
{
    if ((pre&now)!=now) return -1;
    int pos=-1,cnt=0;
    for (int i=0;i<n;i++)
    {
        int x=pre&(1<<i),y=now&(1<<i);
        if (x&&y==0)
        {
            cnt++;
            pos=i;  //pos记录合并后消失的位置
        }
    }
    if (cnt>1) return -1;
    return pos;
}

int dfs(int s)
{
    if (dp[s]!=-1) return dp[s];
    if (s==0) return 0;
    int ans=0;
    for (int i=0;i<(1<<n);i++)
    {
        int pos=isok(s,i);
        if (pos==-1) continue; //不合法，继续找
        int maxx=-INF;
        for (int j=0;j<n;j++)
            if (i&(1<<j))
                maxx=max(maxx,a[j+1][pos+1]+dfs(i));
        ans=max(ans,maxx);
    }
    dp[s]=ans;
    return ans;
}

int main()
{
    int i,j;
    while (~scanf("%d",&n))
    {
        for (i=1;i<=n;i++) for (j=1;j<=n;j++) scanf("%d",&a[i][j]);
        memset(dp,-1,sizeof(dp));
        printf("%d\n",dfs((1<<n)-1));
    }
    return 0;
}