A1146 Topological Order (25) 拓扑排序,vector用法

本题刚开始在做的时候也考虑到了这种方法,但是不知道vector的这种用法,之后参考了柳婼前辈的做法,就做出来了,本题难点在怎么保存入度,以及入度是怎么赋值到临时变量中去的。

#include
#include
#include
using namespace std;
const int maxn=1010;
int in[maxn];
vector<int> v[maxn];
vector<int> ans;
int main()
{
	int n,m,k;
	cin>>n>>m;
	for(int i=0;i<m;i++)
	{
		int a,b;
		scanf("%d %d",&a,&b);
		v[a].push_back(b);
		in[b]++;
	}
	cin>>k;
	for(int i=0;i<k;i++)
	{
		bool flag=false;
		int num;
		vector<int> temp(in,in+n+1);
	    for(int j=0;j<n;j++)
	    {
	    	scanf("%d",&num);
	    	if(temp[num]!=0)flag=true;
	    	for(int l=0;l<v[num].size();l++)
	    	{
	    		temp[v[num][l]]--;
			}
		}
		if(flag)
		{
			ans.push_back(i);
		}
	}
	for(int i=0;i<ans.size();i++)
	{
		printf("%d",ans[i]);
		if(i!=ans.size()-1)
		printf(" ");
	}
}

你可能感兴趣的:(pat刷题)