DFS小结

hdu 2181 哈密顿绕行世界问题

比较直白的一个深搜。

#include<stdio.h>
#include<string.h>
#define N 25
int mark[N][N],vis[N],ans[N];
int m,cnt;
void dfs(int x,int t)
{
    if(t==20&&mark[x][m]==1)
    {
        printf("%d:  ",cnt++);
        for(int i=0;i<=19;i++) printf("%d ",ans[i]);
        printf("%d\n",m);
    }
    for(int i=1;i<=20;i++)
    {
        if(!vis[i]&&mark[x][i])
        {
            ans[t]=i;
            vis[i]=1;
            dfs(i,t+1);
            vis[i]=0;
        }
    }
    return ;
}
int main()
{
    for(int i=1;i<=20;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        mark[i][x]=mark[x][i]=mark[i][y]=mark[y][i]=mark[i][z]=mark[z][i]=1;
    }
    while(scanf("%d",&m),m)
    {
        cnt=1;
        memset(vis,0,sizeof(vis));
        vis[m]=1;
        ans[0]=m;
        dfs(m,1);
    }
    return 0;
}

hdu 2514 Another Eight Puzzle

很好玩儿的一道题，我加了很多特判。

#include<stdio.h>
#include<string.h>
#define N 10
int mark[N][N],vis[N],a[N],b[N];
int ans;
void Init()
{
    memset(mark,0,sizeof(mark));
    mark[1][2]=mark[1][3]=mark[1][4]=1;
    mark[2][1]=mark[2][3]=mark[2][5]=mark[2][6]=1;
    mark[3][1]=mark[3][2]=mark[3][4]=mark[3][5]=mark[3][6]=mark[3][7]=1;
    mark[4][1]=mark[4][3]=mark[4][6]=mark[4][7]=1;
    mark[5][2]=mark[5][3]=mark[5][6]=mark[5][8]=1;
    mark[6][2]=mark[6][3]=mark[6][4]=mark[6][5]=mark[6][7]=mark[6][8]=1;
    mark[7][3]=mark[7][4]=mark[7][6]=mark[7][8]=1;
    mark[8][5]=mark[8][6]=mark[8][7]=1;
    return ;
}
int abs(int x)
{
    if(x<0) return -x;
    return x;
}
void print()
{
    for(int i=1;i<8;i++)
        printf("%d ",b[i]);
    printf("%d\n",b[8]);
    return ;
}
int judge(int k,int x)
{
    for(int i=1;i<=8;i++)
    {
        if(mark[k][i]==1&&a[i]!=0)
            if(abs(x-a[i])==1) return 0;
    }
    return 1;
}
int fun()
{
    for(int i=1;i<=8;i++)
    {
        for(int j=1;j<=8;j++)
        {
            if(mark[i][j]==1&&abs(a[i]-a[j])==1)
                return 0;
        }
    }
    return 1;
}
int dfs(int k)
{
    while(k<=8&&a[k]!=0) k++;
    if(k==9&&fun()) ans++;
    else if(k==9) return 0;
    if(ans==2)
    {
        printf("Not unique\n");
        return 1;
    }
    else if(k==9)
    {
        for(int i=1;i<=8;i++)
            b[i]=a[i];
        return 0;
    }
    for(int i=1;i<=8;i++)
    {
        if(!vis[i]&&judge(k,i))
        {
            a[k]=i;
            vis[i]=1;
            if(dfs(k+1)) return 1;
            vis[i]=0;
            a[k]=0;
        }
    }
    return 0;
}
int main()
{
    int T,cnt=1;
    scanf("%d",&T);
    Init();
    while(T--)
    {
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=8;i++)
        {
            scanf("%d",&a[i]);
            vis[a[i]]=1;
        }
        printf("Case %d: ",cnt++);
        if(a[3]!=1&&a[3]!=8&&a[3]!=0)
        {
            printf("No answer\n");
            continue;
        }
        if(a[6]!=1&&a[6]!=8&&a[6]!=0)
        {
            printf("No answer\n");
            continue;
        }
        if(vis[1]==1&&a[3]!=1&&a[6]!=1)
        {
            printf("No answer\n");
            continue;
        }
        if(vis[8]==1&&a[3]!=8&&a[6]!=8)
        {
            printf("No answer\n");
            continue;
        }
        ans=0;
        dfs(1);
        if(ans==0) printf("No answer\n");
        else if(ans==1) print();
    }
    return 0;
}

hdu 2780 Su-Su-Sudoku

印象里做过不少数独。这道题题意有陷阱，输入的数独有可能直接就是不合格的。

#include<stdio.h>
#include<string.h>
#define N 10
int mark[N][N];
int vis[N];
struct node
{
    int x,y;
} a[N];
int judge(int t,int x,int y)
{
    for(int i=1; i<=9; i++)
    {
        if(mark[i][y]==t||mark[x][i]==t)
            return 0;
    }
    if(x%3==2) x--;
    else if(x%3==0) x-=2;
    if(y%3==2) y--;
    else if(y%3==0) y-=2;
    for(int i=x; i<=x+2; i++)
    {
        for(int j=y; j<=y+2; j++)
            if(mark[i][j]==t)
                return 0;
    }
    return 1;
}
void print()
{
    for(int i=1; i<=9; i++)
    {
        for(int j=1; j<=9; j++)
            printf("%d",mark[i][j]);
        printf("\n");
    }
    return ;
}
int dfs(int k)
{
    if(k==6)
    {
        print();
        return 1;
    }
    else if(k==6) return 0;
    for(int i=1; i<=9; i++)
    {
        if(judge(i,a[k].x,a[k].y))
        {
            mark[a[k].x][a[k].y]=i;
            if(dfs(k+1)) return 1;
            mark[a[k].x][a[k].y]=0;
        }
    }
    return 0;
}
int fun()
{
    for(int i=1; i<=9; i++)
    {
        memset(vis,0,sizeof(vis));
        for(int j=1; j<=9; j++)
        {
            if(mark[i][j]!=0)
            {
                vis[mark[i][j]]++;
                if(vis[mark[i][j]]==2) return 0;
            }
        }
    }
    for(int i=1; i<=9; i++)
    {
        memset(vis,0,sizeof(vis));
        for(int j=1; j<=9; j++)
        {
            if(mark[j][i]!=0)
            {
                vis[mark[j][i]]++;
                if(vis[mark[j][i]]==2) return 0;
            }
        }
    }
    for(int i=1,j=1; i<=7&&j<=7; i+=3,j+=3)
    {
        memset(vis,0,sizeof(vis));
        for(int k=i; k<=i+2; k++)
        {
            for(int l=j; l<=j+2; l++)
            {
                if(mark[k][l]!=0)
                {
                    vis[mark[k][l]]++;
                    if(vis[mark[k][l]]==2) return 0;
                }
            }
        }
    }
    return 1;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int k=1;
        for(int i=1; i<=9; i++)
        {
            for(int j=1; j<=9; j++)
            {
                scanf("%1d",&mark[i][j]);
                if(mark[i][j]==0)
                {
                    a[k].x=i;
                    a[k].y=j;
                    k++;
                }
            }
        }
        if(!fun())
        {
            printf("Could not complete this grid.\n");
            if(T>0) printf("\n");
            continue;
        }
        if(!dfs(1))
            printf("Could not complete this grid.\n");
        if(T>0) printf("\n");
    }
    return 0;
}

hdu 2614 Beat

简单深搜。

#include<stdio.h>
#include<string.h>
#define N 16
int mark[N][N],vis[N];
int max,n;
void dfs(int ans,int f,int k)
{
    if(ans>max) max=ans;
    for(int i=2;i<=n;i++)
    {
        if(!vis[i])
        {
            if(mark[k][i]>=f)
            {
                vis[i]=1;
                dfs(ans+1,mark[k][i],i);
                vis[i]=0;
            }
        }
    }
    return ;
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
                scanf("%d",&mark[i][j]);
        }
        max=1;
        memset(vis,0,sizeof(vis));
        vis[1]=1;
        dfs(1,0,1);
        printf("%d\n",max);
    }
    return 0;
}

hdu 3419 The Three GroupsSticks

对其中几种情况加了特判就过了。

#include<stdio.h>
#include<string.h>
#define N 10
int vis[N],mark[N][N][N];
int a,b,c;
int x,y,z;
int cnt;
void dfs(int t,int f,int k)
{
    if(k==1&&f==a)
        x=t,f=0,k++,t=0;
    else if(k==2&&f==b)
        y=t,f=0,k++,t=0;
    else if(k==3&&f==c)
    {
        z=t;
        if(x*y==z) cnt++;
        return ;
    }
    for(int i=1;i<=9;i++)
    {
        if(vis[i]==0)
        {
            vis[i]=1;
            dfs(t*10+i,f+1,k);
            vis[i]=0;
        }
    }
    return ;
}
int main()
{
    memset(mark,0,sizeof(mark));
    while(scanf("%d%d%d",&a,&b,&c),a+b+c)
    {
        if(a+b<c)
        {
            printf("0\n");
            continue;
        }
        if(a>c||b>c)
        {
            printf("0\n");
            continue;
        }
        if(a+b-c>1)
        {
            printf("0\n");
            continue;
        }
        if(mark[a][b][c])
        {
            printf("%d\n",mark[a][b][c]);
            continue;
        }
        cnt=0;
        memset(vis,0,sizeof(vis));
        dfs(0,0,1);
        mark[a][b][c]=cnt;
        printf("%d\n",mark[a][b][c]);
    }
    return 0;
}

hdu 1455 Sticks

需要大量的剪枝。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define N 70
int a[N],n,aver,vis[N];
int cmp(const void *a,const void *b)
{
    return *(int *)a-*(int *)b;
}
int dfs(int x,int sum,int pos,int f)
{
    if(x==f) return 1;
    for(int i=pos;i<=n;i++)
    {
        int flag=0;
        if(sum+a[i]>aver) return 0;
        if(!vis[i]&&aver-sum>=a[i])
        {
            if(!vis[i-1]&&a[i]==a[i-1]) continue;
            vis[i]=1;
            if(sum+a[i]==aver) flag=dfs(x+1,0,1,f);
            else if(sum+a[i]<aver) flag=dfs(x,sum+a[i],i+1,f);
            vis[i]=0;
            if(flag) return 1;
            if(sum==0) return 0;
            if(sum+a[i]==aver) return 0;
        }
    }
    return 0;
}
int main()
{
    while(scanf("%d",&n),n)
    {
        int sum=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            sum+=a[i];
        }
        int ans=sum;
        qsort(a+1,n,sizeof(a[0]),cmp);
        for(int i=n;i>=2;i--)
        {
            memset(vis,0,sizeof(vis));
            if(sum%i==0)
            {
                aver=sum/i;
                if(a[n]>aver) continue;
                if(dfs(0,0,1,i))
                {
                    ans=aver;
                    break;
                }
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}

hdu 1518 Square

记得写这道题的时候对深搜的剪枝还没有什么概念。。。。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define N 25
int a[N],aver,vis[N];
int n;
int cmp(const void *a,const void *b)
{
    return *(int *)a-*(int *)b;
}
int dfs(int x,int sum,int pos)
{
    if(x==4)
    {
        printf("yes\n");
        return 1;
    }
    for(int i=pos;i<=n;i++)
    {
        if(!vis[i]&&sum+a[i]<=aver)
        {
            vis[i]=1;
            int flag=0;
            if(sum+a[i]==aver) flag=dfs(x+1,0,0);
            else flag=dfs(x,sum+a[i],i+1);
            if(flag) return 1;
            vis[i]=0;
        }
    }
    return 0;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        int sum=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            sum+=a[i];
        }
        if(sum%4!=0)
        {
            printf("no\n");
            continue;
        }
        aver=sum/4;
        memset(vis,0,sizeof(vis));
        qsort(a+1,n,sizeof(a[0]),cmp);
        if(!dfs(0,0,1)) printf("no\n");
    }
    return 0;
}

hdu 1426 Sudoku Killer

需要用scanf("%s",s)读入。

#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
#define N 11
int mark[N][N];
int k;
char s[N];
struct point
{
    int x,y;
} a[N*N];
void print()
{
    for(int i=1; i<=9; i++)
    {
        printf("%d",mark[i][1]);
        for(int j=2; j<=9; j++)
            printf(" %d",mark[i][j]);
        printf("\n");
    }
    return ;
}
int judge(int t,int x,int y)
{
    for(int i=1; i<=9; i++)
    {
        if(mark[i][y]==t||mark[x][i]==t)
            return 0;
    }
    if(x%3==2) x--;
    else if(x%3==0) x-=2;
    if(y%3==2) y--;
    else if(y%3==0) y-=2;
    for(int i=x; i<=x+2; i++)
    {
        for(int j=y; j<=y+2; j++)
           if(mark[i][j]==t)
            return 0;
    }
    return 1;
}
int dfs(int x)
{
    if(x==k)
    {
        print();
        return 1;
    }
    for(int i=1; i<=9; i++)
    {
        if(judge(i,a[x].x,a[x].y))
        {
            mark[a[x].x][a[x].y]=i;
            if(dfs(x+1)==1) return 1;
            mark[a[x].x][a[x].y]=0;
        }
    }
    return 0;
}
int main()
{
    int cnt=0;
    while(scanf("%s",s)!=EOF)
    {
        memset(mark,0,sizeof(mark));
        k=0;
        if(s[0]!='?')
            mark[1][1]=s[0]-'0';
        else
        {
            mark[1][1]=0;
            a[k].x=1;
            a[k].y=1;
            k++;
        }
        for(int i=1; i<=9; i++)
        {
            for(int j=1; j<=9; j++)
            {
                if(i==1&&j==1) continue;
                scanf("%s",s);
                if(s[0]!='?')
                    mark[i][j]=s[0]-'0';
                else
                {
                    mark[i][j]=0;
                    a[k].x=i;
                    a[k].y=j;
                    k++;
                }
            }
            getchar();
        }
        if(cnt!=0) printf("\n");
        cnt++;
        dfs(0);
        //printf("Over\n");
        getchar();
    }
    return 0;
}

hdu 4431 Mahjong

写了很多遍才写对，就是因为做这道题的时候一直错我才专门练习了DFS。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define N 35
int flag[N],mark[N],ans[N];
int cmp(const void *a,const void *b)
{
    return *(int *)a-*(int *)b;
}
int judge()
{
    int flag=1;
    for(int i=1; i<=34; i++) if(mark[i]%2!=0||mark[i]==4) flag=0;
    if(flag) return 1;
    flag=1;
    int cnt=0;
    if(mark[1]==0||mark[9]==0||mark[10]==0||mark[18]==0||mark[19]==0||mark[27]==0) flag=0;
    for(int i=28; i<=34; i++) if(mark[i]==0) flag=0;
    if(mark[1]==2||mark[9]==2||mark[10]==2||mark[18]==2||mark[19]==2||mark[27]==2) cnt=1;
    for(int i=28; i<=34; i++) if(mark[i]==2) cnt=1;
    if(flag==0) return 0;
    if(cnt==0) return 0;
    return flag;
}
int dfs(int pos,int k,int flag)
{
    if(k==4&&flag==1) return 1;
    for(int i=pos;i<=34;i++)
    {
        if(mark[i]>=3)
        {
            mark[i]-=3;
            if(dfs(i,k+1,flag))
            {
                mark[i]+=3;
                return 1;
            }
            mark[i]+=3;
        }
        if(mark[i]>=2&&flag==0)
        {
            mark[i]-=2;
            if(dfs(i,k,1))
            {
                mark[i]+=2;
                return 1;
            }
            mark[i]+=2;
        }
        if(mark[i]>=1&&i<=27)
        {
            if(i%9==1)
            {
                if(mark[i+1]&&mark[i+2])
                {
                    mark[i]--;mark[i+1]--;mark[i+2]--;
                    if(dfs(i+1,k+1,flag))
                    {
                        mark[i]++;mark[i+1]++;mark[i+2]++;
                        return 1;
                    }
                    mark[i]++;mark[i+1]++;mark[i+2]++;
                }
            }
            else if(i%9==0)
            {
                if(mark[i-1]&&mark[i-2])
                {
                    mark[i-1]--;mark[i]--;mark[i-2]--;
                    if(dfs(i+1,k+1,flag))
                    {
                        mark[i-1]++;mark[i]++;mark[i-2]++;
                        return 1;
                    }
                    mark[i-1]++;mark[i]++;mark[i-2]++;
                }
            }
            else if(i%9==2)
            {
                if(mark[i-1]&&mark[i+1])
                {
                    mark[i-1]--;mark[i]--;mark[i+1]--;
                    if(dfs(i+1,k+1,flag))
                    {
                        mark[i-1]++;mark[i]++;mark[i+1]++;
                        return 1;
                    }
                    mark[i-1]++;mark[i]++;mark[i+1]++;
                }
                if(mark[i+1]&&mark[i+2])
                {
                    mark[i]--;mark[i+1]--;mark[i+2]--;
                    if(dfs(i+1,k+1,flag))
                    {
                        mark[i]++;mark[i+1]++;mark[i+2]++;
                        return 1;
                    }
                    mark[i]++;mark[i+1]++;mark[i+2]++;
                }
            }
            else if(i%9==8)
            {
                if(mark[i-1]&&mark[i+1])
                {
                    mark[i-1]--;mark[i]--;mark[i+1]--;
                    if(dfs(i+1,k+1,flag))
                    {
                        mark[i-1]++;mark[i]++;mark[i+1]++;
                        return 1;
                    }
                    mark[i-1]++;mark[i]++;mark[i+1]++;
                }
                if(mark[i-1]&&mark[i-2])
                {
                    mark[i-1]--;mark[i]--;mark[i-2]--;
                    if(dfs(i+1,k+1,flag))
                    {
                        mark[i-1]++;mark[i]++;mark[i-2]++;
                        return 1;
                    }
                    mark[i-1]++;mark[i]++;mark[i-2]++;
                }
            }
            else
            {
                if(mark[i-1]&&mark[i-2])
                {
                    mark[i-1]--;mark[i]--;mark[i-2]--;
                    if(dfs(i+1,k+1,flag))
                    {
                        mark[i-1]++;mark[i]++;mark[i-2]++;
                        return 1;
                    }
                    mark[i-1]++;mark[i]++;mark[i-2]++;
                }
                if(mark[i-1]&&mark[i+1])
                {
                    mark[i-1]--;mark[i]--;mark[i+1]--;
                    if(dfs(i+1,k+1,flag))
                    {
                        mark[i-1]++;mark[i]++;mark[i+1]++;
                        return 1;
                    }
                    mark[i-1]++;mark[i]++;mark[i+1]++;
                }
                if(mark[i+1]&&mark[i+2])
                {
                    mark[i]--;mark[i+1]--;mark[i+2]--;
                    if(dfs(i+1,k+1,flag))
                    {
                        mark[i]++;mark[i+1]++;mark[i+2]++;
                        return 1;
                    }
                    mark[i]++;mark[i+1]++;mark[i+2]++;
                }
            }
        }
        if(mark[i]) return 0;
    }
    return 0;
}
int main()
{
    int T;
    scanf("%d",&T);
    getchar();
    while(T--)
    {
        memset(flag,0,sizeof(flag));
        memset(mark,0,sizeof(mark));
        for(int i=1; i<=13; i++)
        {
            int x;
            char c;
            scanf("%d%c",&x,&c);
            if(c=='m') x=x;
            else if(c=='s') x+=9;
            else if(c=='p') x+=18;
            else x+=27;
            mark[x]++;
            flag[x]++;
        }
        int k=0;
        for(int i=1; i<=34; i++)
        {
            if(flag[i]<=3) mark[i]++;
            else continue;
            if(judge())
            {
                ans[k++]=i;
                mark[i]--;
                continue;
            }
            if(dfs(1,0,0)) ans[k++]=i;
            mark[i]--;
        }
        if(k==0) printf("Nooten\n");
        else
        {
            printf("%d ",k);
            for(int i=0; i<k-1; i++)
            {
                if(ans[i]/9==0||(ans[i]/9==1&&ans[i]%9==0)) printf("%dm ",ans[i]%9==0?9:ans[i]%9);
                else if(ans[i]/9==1||(ans[i]/9==2&&ans[i]%9==0)) printf("%ds ",ans[i]%9==0?9:ans[i]%9);
                else if(ans[i]/9==2||(ans[i]/9==3&&ans[i]%9==0)) printf("%dp ",ans[i]%9==0?9:ans[i]%9);
                else printf("%dc ",ans[i]%9==0?9:ans[i]%9);
            }
            if(ans[k-1]/9==0||(ans[k-1]/9==1&&ans[k-1]%9==0)) printf("%dm\n",ans[k-1]%9==0?9:ans[k-1]%9);
            else if(ans[k-1]/9==1||(ans[k-1]/9==2&&ans[k-1]%9==0)) printf("%ds\n",ans[k-1]%9==0?9:ans[k-1]%9);
            else if(ans[k-1]/9==2||(ans[k-1]/9==3&&ans[k-1]%9==0)) printf("%dp\n",ans[k-1]%9==0?9:ans[k-1]%9);
            else printf("%dc\n",ans[k-1]%9==0?9:ans[k-1]%9);
        }
    }
    return 0;
}
/*
6
2m 3m 1p 1p 1p 2p 2p 2p 3p 3p 3p 4c 4c
1m 9m 1s 9s 1p 9p 1c 2c 3c 4c 5c 6c 7c
1m 1m 1p 1p 2p 2p 3p 3p 4p 4p 5p 5p 6p
1s 2s 3s 2c 2c 2c 2p 3p 5m 6m 7m 1p 1p
1p 1p 2p 3p 4s 5s 6s 7c 7c 3s 3s 2m 2m
1p 1p 2p 2p 3p 3p 4p 4p 5p 5p 6p 6p 7p
答案：
2 1m 4m
13 1m 9m 1s 9s 1p 9p 1c 2c 3c 4c 5c 6c 7c
2 3p 6p
2 1p 4p
Nooten
3 1p 4p 7p
*/