ZOJ 1204 Additive equations

纠结了很久的一道搜索题目

提交了10次才通过。

一开始是TLE，然后就一直是WA了。

最后实在找不到原来的错误，就决定重敲，竟然异常地顺利，1A过了。

而且重敲的代码也比原来的代码简洁很多。

当局面已经乱得一塌糊涂的时候，不如快刀斩乱麻。推倒重建反倒是更优的策略。不过还是会有些纠结没有把原来的错误找出来。

先贴上重敲然后AC的代码

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;

#define MAXN 33
int s[MAXN];
int visit[MAXN];
int ct;
int n;
bool is_out;
//td目标因子数，cd现在因子数，pos现在位置，sum现在总和
void dfs(int td,int cd,int pos,int sum)
{
    if (td==cd){
        for (int i=pos;i<n;i++){
            if (sum==s[i]){
                is_out=true;
                printf("%d", visit[0]);
                for (int j=1;j<ct;j++){
                    printf("+%d", visit[j]);
                }
                printf("=%d\n", s[i]);
                //很诡异的一个剪枝，我注释掉之后竟然还快乐10ms，估计是服务器问题
                break;
            }
        }
        return;
    }
    for (int i=pos;i<n+cd-td;i++){
        //最关键的一个剪枝，从10001ms超时变成70ms
        if (sum+s[i]>s[n-1]) return;
        visit[ct++] = s[i];
        dfs(td,cd+1,i+1,sum+s[i]);
        ct--;
    }
    return;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        is_out=false;
        ct = 0;
        for (int i=0;i<n;i++){
            scanf("%d",&s[i]);
        }
        sort(s,s+n);
        //目的因子数
        for (int i=2;i<n;i++){
            dfs(i,0,0,0);
        }
        if (is_out==false)
            printf("Can't find any equations.\n");
        printf("\n");
    }
}

然后再贴一下一直WA的悲剧的代码

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <cstring>
#include <algorithm>

using namespace std;

#define N 31

struct basic
{
    int index;
    int sum;
    string result;
    basic()
    {
        index = 0;
        sum = 0;
        result = "";
    }
    basic(const basic& rhs)
    {
        index = rhs.index;
        sum = rhs.sum;
        result = rhs.result;
    }
};

struct node
{
    string prt;
    int no;
};

bool cmp(const node &a, const node &b){
    if (a.prt.length()<b.prt.length()) return true;
    if (a.prt.length()==b.prt.length())
        return a.no<b.no;
    return false;
}


int s[N];int m;
bool is_out;
int cou ;
node result[30000];
void dfs(basic now)
{
    if (now.sum>s[m-1]) return;
    for (int i=now.index+1;i<m;i++)
    {
        if (now.sum+s[now.index]==s[i])
        {
            is_out = true;
            char t[256];
            string topu;
            sprintf(t, "+%d=%d", s[now.index],s[i]);
            topu = t;
            result[cou].no = cou;
            result[cou++].prt =now.result+topu;
            break;
        }
        else
            if (now.sum+s[now.index]<s[i]) break;
    }
    if (now.index >= m-1) return;
    basic next(now);
    next.index++;

    next.sum += s[now.index];
    char t[256];
    string topu;
    sprintf(t, "%d", s[now.index]);
    topu = t;
    if (now.sum!=0)
        next.result += "+" +topu;
    else
        next.result = topu;
    dfs(next);

    next.result = now.result;
    next.sum -=  s[now.index];
    dfs(next);
}


int main()
{
    int n;
    scanf("%d",&n);
    while(n--)
    {
        scanf("%d",&m);
        memset(s,0,sizeof(s));
        cou = 0;
        is_out = false;
        for(int i=0;i<m;i++)
        {
            scanf("%d",&s[i]);
        }
        sort(s,s+m);
        basic now ;
        dfs(now);
        sort(result,result+cou,cmp);
        for ( int i=0;i<cou;i++)
        {
            printf("%s\n",(result[i].prt).c_str());
        }
//        printf("%d\n",cou);
        if (!is_out) printf("Can't find any equations.\n");
        printf("\n");
    }

    return 0;
}