题意:
给定一串数字,在这些数字中插入一些‘+’或者’=’ 使得两边相等
思路:
比较简单的dfs问题
首先确定下等于号的位置,然后两边进行dfs。
my code
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N = 20;
char num[N];
int len, lsum, rsum;
int ans;
int trans(int st, int en) {
int res = 0;
for(int i = st; i <= en; i++)
res = res * 10 + (num[i] - '0');
return res;
}
void dfsr(int st, int en, int sum) {
if(st > en) {
if(sum == lsum) ans++;
return ;
}
for(int i = st; i <= en; i++) {
int tmp = trans(st, i);
dfsr(i+1, en, sum + tmp);
}
}
void dfsl(int st, int en, int sum) {
if(st > en) {
lsum = sum;
dfsr(en+1, len, 0);
return ;
}
for(int i = st; i <= en; i++) {
int tmp = trans(st, i);
dfsl(i+1, en, sum + tmp);
}
}
int main() {
while(~scanf("%s", num+1) && num[1] != 'E') {
len = strlen(num+1);
ans = 0;
for(int i = 1; i < len; i++) {
dfsl(1, i, 0);
}
printf("%d\n", ans);
}
return 0;
}