hdu 4403 dfs

巨麻烦的模拟题orz。。。。

先确定等号的位置，然后两层dfs，分别算等号前面的和后面的，再比较

话说这题竟然不开long long都能水过

 

#include <iostream>
#include<cstring>
using namespace std;
int ANS,ansx,ansy,T,l;
string s;
int nx[1000],ny[1000],ax[1000],ay[1000],a[1000];
int lx,ly;

int calcx()
{
    //ax[1..lx-1]
    //nx[1..lx]
    int tmpn=nx[1];
    int tmpa=0;
    for(int i=1;i<=lx-1;i++)
    {
        if(ax[i]==1)
        {
            tmpa+=tmpn;
            tmpn=nx[i+1];
        }
        else
        {
            tmpn=tmpn*10+nx[i+1];
        }
    }
    tmpa+=tmpn;
    return tmpa;
}

int calcy()
{
    //ay[1..ly-1]
    //ny[1..ly]
    int tmpn=ny[1];
    int tmpa=0;
    for(int i=1;i<=ly-1;i++)
    {
        if(ay[i]==1)
        {
            tmpa+=tmpn;
            tmpn=ny[i+1];
        }
        else
        {
            tmpn=tmpn*10+ny[i+1];
        }
    }
    tmpa+=tmpn;
    return tmpa;
}

void dfsy(int y)
{
    if(y==ly)
    {
        //cout<<nx[1]<<" ";
        //for(int i=1;i<=lx-1;i++)  cout<<char(ax[i]+'A')<<" "<<nx[i+1]<<" ";
        //cout<<"----";
        //cout<<ny[1]<<" ";
        //for(int i=1;i<=ly-1;i++)  cout<<char(ay[i]+'A')<<" "<<ny[i+1]<<" ";

        ansy=calcy();
        //cout<<"  "<<ansx<<"=="<<ansy<<" ";
        if(ansy==ansx)
        {
            ANS++;
        //    cout<<" OK! ";
        }
        //cout<<endl;
    }
    else
    {
        ay[y]=1;
        dfsy(y+1);
        ay[y]=0;
        dfsy(y+1);
    }
}

void dfsx(int x)
{
    if(x==lx)
    {
        ansx=calcx();
        dfsy(1);
    }
    else
    {
        ax[x]=1;        //add a plus after ax[x]
        dfsx(x+1);
        ax[x]=0;
        dfsx(x+1);
    }
}

int main()
{
    while(cin>>s)
    {
        if(s=="END")    break;
        l=s.length();
        ANS=0;  ansx=0;  ansy=0;
        for(int i=1;i<=l;i++)
            a[i]=s[i-1]-'0';
        for(int eq=1;eq<=l-1;eq++)     //add equal after a[eq]
        {
            memset(nx,0,sizeof(nx));
            memset(ny,0,sizeof(ny));
            for(int i=1;i<=eq;i++)      nx[i]=a[i];
            lx=eq;
            for(int i=eq+1;i<=l;i++) ny[i-eq]=a[i];
            ly=l-eq;
            dfsx(1);
        }
        cout<<ANS<<endl;
    }

    return 0;
}