【版本一:无脑DFS,27分,最后一个测试点提示段错误,一脸懵逼中】
#include <bits/stdc++.h>
using namespace std;
struct Node
{
int weight;
int last;
vector<int> next;
};
Node node[101];
vector<vector<int> >ans;
bool cmp(vector<int> a,vector<int> b)
{
for(int i=0;i<a.size();i++)
if(a[i]!=b[i]) return a[i]>b[i];
}
int main(void)
{
int n,m,S,sum=0;
vector<int> temp;
stack<pair<int,int> > s; //nodeID and weight
pair<int,int> pre(0,0);
scanf("%d%d%d",&n,&m,&S);
for(int i=0;i<n;i++)
scanf("%d",&node[i].weight);
node[0].last=-1;
for(int i=0;i<m;i++)
{
int pos,num,tmp;
scanf("%d%d",&pos,&num);
for(int j=0;j<num;j++)
{
scanf("%d",&tmp);
node[pos].next.push_back(tmp);
node[tmp].last=pos;
}
}
s.push(pre);
while(!s.empty())
{
pre=s.top();
s.pop();
sum=node[pre.first].weight + pre.second;
if(sum == S&&node[pre.first].next.size()==0) //this is one possible answer
{
int i=pre.first;
while(i!=-1)
{
temp.push_back(node[i].weight);
i=node[i].last;
}
reverse(temp.begin(),temp.end());
ans.push_back(temp);
temp.clear();
continue;
}
else if(sum>=S) continue;
else
{
for(int i=0;i<node[pre.first].next.size();i++)
{
pair<int,int> tmp(node[pre.first].next[i], sum);
s.push(tmp);
}
}
}
sort(ans.begin(),ans.end(),cmp);
for(int i=0;i<ans.size();i++)
{
for(int j=0;j<ans[i].size();j++)
{
printf("%d",ans[i][j]);
if(j!=ans[i].size()-1)
printf(" ");
}
if(i!=ans.size()-1)
printf("\n");
}
}
【版本二:令人窒息的骚操作,30分~】
我自己实现的cmp函数,必须能比较出两条路径相等的情况,而且必须return false!才能通过,否则第五个测试点会显示段错误。坑点!!
代码没有什么不同之处,就是加了个return false;仅此而已。此题过~
#include <bits/stdc++.h>
using namespace std;
struct Node
{
int weight;
int last;
vector<int> next;
};
Node node[101];
vector<vector<int> >ans;
bool cmp(vector<int> a,vector<int> b)
{
for(int i=0;i<a.size();i++)
if(a[i]!=b[i]) return a[i]>b[i];
return 0;
}
int main(void)
{
int n,m,S,sum=0;
vector<int> temp;
stack<pair<int,int> > s; //nodeID and weight
pair<int,int> pre(0,0);
scanf("%d%d%d",&n,&m,&S);
for(int i=0;i<n;i++)
scanf("%d",&node[i].weight);
node[0].last=-1;
for(int i=0;i<m;i++)
{
int pos,num,tmp;
scanf("%d%d",&pos,&num);
for(int j=0;j<num;j++)
{
scanf("%d",&tmp);
node[pos].next.push_back(tmp);
node[tmp].last=pos;
}
}
s.push(pre);
while(!s.empty())
{
pre=s.top();
s.pop();
sum=node[pre.first].weight + pre.second;
if(sum == S&&node[pre.first].next.size()==0) //this is one possible answer
{
int i=pre.first;
while(i!=-1)
{
temp.push_back(node[i].weight);
i=node[i].last;
}
reverse(temp.begin(),temp.end());
ans.push_back(temp);
temp.clear();
continue;
}
else if(sum>=S) continue;
else
{
for(int i=0;i<node[pre.first].next.size();i++)
{
pair<int,int> tmp(node[pre.first].next[i], sum);
s.push(tmp);
}
}
}
sort(ans.begin(),ans.end(),cmp);
for(int i=0;i<ans.size();i++)
{
for(int j=0;j<ans[i].size();j++)
{
printf("%d",ans[i][j]);
if(j!=ans[i].size()-1)
printf(" ");
}
if(i!=ans.size()-1)
printf("\n");
}
}