HDU4403-模拟、数学

一道很难的奥数题，给出一个数字串，插入加号和等号使之成立。求成立的算式数。

我的做法是，先分成两段，中间插入等号 ，再分别求出左右两边可能的值和个数，然后对比，把值相等的情况乘起来，加到最终结果上。

求可能值用递归，每次遇到一个数就选择是插入加号还是不插入，不插入加号就把之前维护的值乘以10加上此值，插入加号把维护的值加到和里。这样可以把所有的情况存进map里。

#include <algorithm>
#include <cstring>
#include <ctype.h>
#include <cstdlib>
#include <cstdio>
#include <vector>
#include <string>
#include <queue>
#include <stack>
#include <cmath>
#include <set>
#include <map>

using namespace std;

int N,M,T;
char save[20];

map <long long,int> m[2];

void solve(int l,int r,long long cur,long long combo,int d)
{
if(l >= r)
{
cur += combo;
if(m[d].count(cur) == 0)
m[d].insert( pair<long long,int>(cur,1));
else m[d][cur]++;
return;
}

solve(l+1,r,cur+combo,save[l+1]-'0',d);
solve(l+1,r,cur,combo*10+save[l+1]-'0',d);
return;
}

int main()
{
while(scanf("%s",save) && save[0] != 'E')
{
int n = strlen(save);
int ans = 0;
for(int i = 0;i < n-1;i++)
{
m[0].clear();m[1].clear();
solve(0,i,0,save[0]-'0',0);
solve(i+1,n-1,0,save[i+1]-'0',1);
map<long long,int>::iterator it = m[0].begin();

for(;it != m[0].end();it++)
{
//printf("%d %d\n",it->first,it->second);
if(m[1].count(it->first) )
ans += (it->second * m[1][it->first]);
}
}
printf("%d\n",ans);
}
}