【PAT 甲级】A1053 Path of Equal Weight（30分）

【版本一：无脑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");
	}
}