HDU 4403 A very hard Aoshu problem

回溯,两个数字之间标记0,1,最后将标记1的位置放‘+’号,‘0’的位置处两边数字合成一个数,最后将每两个数之间放等号,检查是否相等。

代码如下:

#include <iostream>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
using namespace std;
int len, num[16], vis[16], cct;
char a[16];
void dfs(int cur)
{
    if(cur == len-1)
    {
        int bei, suml, sumr;
        int ct = 0, f1 = 0;
        for(int i = 0; i < len; i++) // 将各个数字合成一个完整的数
        {
            if(vis[i])
            {
                num[ct] = 0;
                bei = 1;
                for(int j = i; j >= f1; j--)
                {
                    num[ct] += (a[j] - '0') * bei;
                    bei *= 10;
                }
                f1 = i + 1;
                ++ct;
            }
        }
        if(f1 != len) // 保证最后几个数字也可以合成一个完整的数
        {
            num[ct] = 0;
            bei = 1;
            for(int j = len - 1; j >= f1; j--)
            {
                num[ct] += (a[j] - '0') * bei;
                bei *= 10;
            }
            ++ct;
        }
        for(int k = 0; k < ct; k++) // 检查两边数字之和是否相等
        {
            suml = 0, sumr = 0;
            for(int i = 0; i < k; i++)
                suml += num[i];
            for(int i = k; i < ct; i++)
                sumr += num[i];
            if(suml == sumr)
                ++cct;
        }
        return ;
    }
    vis[cur] = 1;
    dfs(cur + 1);
    vis[cur] = 0;
    dfs(cur + 1);
}
int main()
{
    memset(vis, 0, sizeof(vis));
    while(scanf("%s", a)!=EOF)
    {
        cct = 0;
        if(!strcmp(a, "END"))
            break;
        len =  strlen(a);
        vis[len] = 1;
        dfs(0);
        printf("%d\n", cct);
    }
}


你可能感兴趣的:(HDU 4403 A very hard Aoshu problem)