hdu 4403 A very hard Aoshu problem(dfs)

题意:

给定一串数字,在这些数字中插入一些‘+’或者’=’ 使得两边相等

思路:

比较简单的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;
}

你可能感兴趣的:(HDU,4403)