纠结了很久的一道搜索题目
提交了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;
}