hdu 4090 GemAnd Prince dfs+bfs+最优性剪枝

思路，dfs搜索，最优性剪枝。

个人被坑很久的地方就是枚举格子的时候，应该是枚举整个八连快，不然的话一定超时。

代码有点丑，将就一下吧。。。

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<queue>#include<vector>#include<cmath>#include<utility>#include<map>#include<string>using namespace std;int a[10][10];int n,m,k;int sum;int ans;int cnt[10];int dx[]={-1,-1,-1,0,0,1,1,1};int dy[]={-1,0,1,-1,1,-1,0,1};struct node{    int x,y;}Q[10000];int top;int isok(int x,int y,int use[][10]){    top=0;    node t,f;    int front=0,rear=0;    f.x=x;f.y=y;    Q[rear++]=f;    use[x][y]=1;    while(front<rear)    {        f=Q[front++];        for(int d=0;d<8;d++)        {            t=f;            t.x+=dx[d];            t.y+=dy[d];            if(a[t.x][t.y]==a[x][y]&&!use[t.x][t.y])            {               use[t.x][t.y]=1;               Q[rear++]=t;            }        }    }    top=rear;    return rear;}inline void save(int sa[][10]){    for(int i=1;i<=n;i++)        for(int j=1;j<=m;j++)            sa[i][j]=a[i][j];}inline void read(int sa[][10]){    for(int i=1;i<=n;i++)        for(int j=1;j<=m;j++)            a[i][j]=sa[i][j];}void debug(){    for(int i=1;i<=n;i++){        for(int j=1;j<=m;j++)            cout<<a[i][j];            cout<<endl;    }    system("pause");}void trans(){    for(int i=0;i<top;i++)        a[Q[i].x][Q[i].y]=0;    for(int i=1;i<=n-1;i++)        for(int j=1;j<=m;j++)        {            int x=i,y=j;            if(a[i][j]&&!a[i+1][j])            {                while(a[x][y])                {                    a[x+1][y]=a[x][y];                    x--;                }                a[x+1][y]=0;            }        }    for(int j=m;j>=1;j--)    {        int ok=1;        for(int i=1;i<=n;i++)            if(a[i][j]) ok=0;        if(ok)        {            int p,k;            for(int q=1;q<=n;q++)            {                for(p=m;p>j;p--)                    if(a[q][p]) break;                if(p==j) continue;                for(k=j;k<p;k++)                    a[q][k]=a[q][k+1];                a[q][k]=0;            }        }    }}inline int h(){    int tot=0;    for(int i=1;i<=k;i++)    {        if(cnt[i]>=3) tot+=cnt[i]*cnt[i];    }    return tot;}void dfs(int score){    ans=max(ans,score);    if(h()+score<=ans) return;    int tmp;    int use[10][10]={0};    for(int i=1;i<=n;i++)        for(int j=1;j<=m;j++)        {            if(a[i][j]==0&&!use[i][j]) continue;            tmp=isok(i,j,use);            if(tmp>=3)            {                int sa[10][10];                cnt[a[i][j]]-=tmp;                save(sa);                trans();                dfs(score+tmp*tmp);                read(sa);                cnt[a[i][j]]+=tmp;            }        }}int main(){    while(scanf("%d%d%d",&n,&m,&k)!=EOF)    {        sum=0;        memset(a,0,sizeof(a));        memset(cnt,0,sizeof(cnt));        for(int i=1;i<=n;i++)        {            for(int j=1;j<=m;j++)            {                scanf("%d",&a[i][j]);                cnt[a[i][j]]++;            }        }        ans=0;        dfs(0);        printf("%d\n",ans);    }    return 0;}