HDU 4403 A very hard Aoshu problem(暴搜)

题意:在一个数字串里加入加号,等号,使等式成立。有几种方法;

思路:暴搜:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
using namespace std;
map<long long,int> mp;
int ans;
int a[39],len,cnt;
void dfs1(long long s,long long t,int c,int k)
{
    if(c==k)
    {
        if(mp.find(s+t)==mp.end())
        mp[s+t]=1;
        else
        mp[s+t]++;
        return ;
    }
    dfs1(s+t,a[c],c+1,k);
    if(t>0ll)
    dfs1(s,t*10+a[c],c+1,k);
}
void dfs2(long long s,long long t,int c,int k)
{
    if(c==k)
    {
        if(mp.find(s+t)!=mp.end())
        ans+=mp[s+t];
        return ;
    }
    dfs2(s+t,a[c],c+1,k);
    if(t>0ll)
    dfs2(s,t*10+a[c],c+1,k);
}

int main()
{
    freopen("in.txt","r",stdin);
    char str[39];
    while(~scanf("%s",str))
    {
        if(str[0]=='E') break;
        len = strlen(str);
        for(int i=0;i<len;i++)
        a[i]=str[i]-'0';
        ans=0;
        for(int i=1;i<len;i++)
        {
            mp.clear();cnt =1;
            dfs1(0,0,0,i);
            dfs2(0,0,i,len);
        }
        printf("%d\n",ans);
    }
    return 0;
}


你可能感兴趣的:(c)