poj 3548 Restoring the digits(DFS)

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int use[30],a[30],flag,b[30],m,p[20];
char s[1024];

void dfs(int x)
{
    int i,a1,a2,a3,t,l;
    if(flag) return;
    if(x==m)
    {
        l=strlen(s);
        a1=0;
        a2=0;
        a3=0;
        for(i=0; i<l; i++)
        {
            if(s[i]=='+')
            {
                t=1;
                break;
            }
            else if(s[i]=='-')
            {
                t=-1;
                break;
            }

            if(s[i]>='0'&&s[i]<='9')
            {
                a1=a1*10+s[i]-'0';
            }
            else if(s[i]<='Z'&&s[i]>='A')
            {
                a1=a1*10+a[s[i]-'A'];
            }
        }
        for(i=i+1; s[i]!='='; i++)
        {
            if(s[i]>='0'&&s[i]<='9')
            {
                a2=a2*10+s[i]-'0';
            }
            else if(s[i]<='Z'&&s[i]>='A')
            {
                a2=a2*10+a[s[i]-'A'];
            }
        }
        for(i=i+1; i<l; i++)
        {
            if(s[i]>='0'&&s[i]<='9')
            {
                a3=a3*10+s[i]-'0';
            }
            else if(s[i]<='Z'&&s[i]>='A')
            {
                a3=a3*10+a[s[i]-'A'];
            }
        }
        if(t==1&&a1+a2==a3) flag=1;
        else if(t==-1&&a1-a2==a3) flag=1;
        return ;
    }
    for(i=0; i<=9; i++)
    {
        if(p[i]==1) continue;
        p[i]=1;
        a[b[x]]=i;
        dfs(x+1);
        if(flag) return;
        p[i]=0;
    }
    return ;
}

int main()
{
    int i,j;
    while(~scanf("%s",s))
    {
        int len=strlen(s);
        memset(use,0,sizeof(use));
        memset(p,0,sizeof(p));
        flag=0;
        for(i=0; i<len; i++)
        {
            if(s[i]<='Z'&&s[i]>='A')
            {
                use[s[i]-'A']=1;
            }
        }

        for(i=0,m=0;i<26;i++)
            if(use[i]==1) b[m++]=i;

        dfs(0);
        for(i=0; i<m; i++)
         printf("%c %d\n",b[i]+'A',a[b[i]]);

    }
    return 0;
}

你可能感兴趣的:(poj 3548 Restoring the digits(DFS))