JAG Summer 2012 Day 4 C Connect

状压dp，由于枚举两维状态会GG，所以只枚举当前位置前m个的状态，就是这个样子大概= =；

呆马：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <vector>
#include <queue>
#define inf 1000000007
#define maxn 130
#define maxm 18

using namespace std;

int f[2][maxm][65538];
int bit[65538][maxm];
char s[maxn][maxm];
int len[maxn];
int n,m,mxbit,hi;

int up(int x, int y, int mask)
{
    int cnt=len[x-1], now=0;;
    for (int i=y;i<m;i++) cnt-=bit[mask][i];
    for (int i=1;i<y;i++) now+=bit[mask][i];
    if (bit[mask][m] && s[x-1][cnt-1]==s[x][now]) return 1;
    return 0;
}

int lf(int x, int y, int mask)
{
    int now=0;
    for (int i=1;i<y;i++) now+=bit[mask][i];
    if (bit[mask][1] && s[x][now-1]==s[x][now]) return 1;
    return 0;
}

int cal(int y, int mask)
{
    int now=0;
    for (int i=1;i<=y;i++) now+=bit[mask][i];
    return now;
}

void get_mask()
{
    for (int i=1;i<=mxbit;i++)
    {
        int now=i;
        for (int j=1;j<=m;j++)
        {
            bit[i][j]=now%2;
            now/=2;
        }
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    mxbit=(1<<m)-1;
    get_mask();
    for (int i=1;i<=n;i++) 
    {
        scanf("%s",s[i]);
        len[i]=strlen(s[i]);
    }
    for (int i=0;i<=1;i++)
        for (int j=1;j<=m;j++)
        {
            for (int mask=0;mask<=mxbit;mask++)
                f[i][j][mask]=-inf;
        }
    f[1][1][0]=f[1][1][1]=0;
    for (int i=1;i<=n;i++)
    {
        int p=i&1;
        for (int j=1;j<=m;j++)
            for (int mask=0;mask<=mxbit;mask++)
            {
                int tmp=(mask<<1)&mxbit;
                if (i==1 && j==1) continue;
                if (j==1)
                {    
                    if (cal(m,mask)!=len[i-1]) continue;
                    f[p][j][tmp]=max(f[p][j][tmp],f[p^1][m][mask]);
                    f[p][j][tmp|1]=max(f[p][j][tmp|1],f[p^1][m][mask]+2*up(i,j,mask));
                }
                else
                {
                    if (cal(j-1,mask)>len[i]) continue;
                    f[p][j][tmp]=max(f[p][j][tmp],f[p][j-1][mask]);
                    if (cal(j-1,mask)<len[i])
                        f[p][j][tmp|1]=max(f[p][j][tmp|1],f[p][j-1][mask]+2*lf(i,j,mask)+2*up(i,j,mask));

                }
            }
    }
    int ans=0;
    for (int mask=0;mask<=mxbit;mask++)
        if (cal(m,mask)==len[n])
            ans=max(ans,f[n&1][m][mask]);
    printf("%d
",ans);
    return 0;
}