Codeforces Round #895 (Div. 3)F. Selling a Menagerie(拓扑排序，贪心)

原题链接:F. Selling a Menagerie
题目大意:
动物园里有n个动物，第i个动物害怕第 a_i个动物，第i个动物价值 c_i元。现在我要将这些动物全部卖掉。
若买出i之前未卖出a_i，则获得的价格将乘以2，也就是 2×c_i。
求出卖出顺序。
数据范围：1≤t≤10⁴ ，2≤n≤10⁵，1≤a_i≤n , a_i≠i，1≤c_i≤10⁹，
保证所有测试用例的n之和不超过10⁵
解题思路：
根据贪心的思想，对于不被任何动物害怕的动物，我们可以最先卖出它们，对于剩下来的动物，我们可以重复这个过程，先把害怕它的动物卖掉，最后再卖掉它
再实际操作过程中，我们可以建立一个图，假设编号为i的点的入度为害怕i号动物的数量，可以建出类似下图的图：

首先对于非环上的点（红色的点），我们可以先把入读为0的点卖掉，然后使它指向的点入读减一，重复这个过程，就可以把非环上的点都卖掉，可以采用拓扑排序的做法：

for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		du[a[i]]++;//记录入度 
	}
	for(int i=1;i<=n;i++)cin>>c[i];
	queue<int> q;
	vector<int> ans;
	//拓扑排序 
	for(int i=1;i<=n;i++)
	{
		if(!du[i])q.push(i);//入读为0的点入队 
	}
	while(!q.empty())
	{
		int k=q.front();
		q.pop();
		vis[k]=1;
		ans.push_back(k);
		du[a[k]]--;//删去一个点后，使它指向的点入度减一 
		if(!du[a[k]])q.push(a[k]);
	}

对于环上的点（蓝色的点），可以采用贪心的思想，把价格最小的动物标记，然后从它害怕的动物开始卖，最后卖掉它，这样环上除了价格最小的动物外，其它动物均可以以两倍价格卖出，得到最大收入：

void dfs(int i)
{
	if(vis[i])return;
	vis[i]=1;
	if(c[a[i]]<mi)
	{
		mi=c[a[i]];
		id=a[i];
	}
	dfs(a[i]);
}
for(int i=1;i<=n;i++)
	{
		if(!vis[i])
		{
			mi=c[i];
			id=i;
			dfs(i);//找到价格最小的动物 
			int x=a[id];
			//卖掉环上的动物 
			do
			{
				ans.push_back(x);
				x=a[x];
			}while(x!=a[id]);
		}
	}

AC代码：

#include<bits/stdc++.h>
using namespace std;
#define int long long 
int a[500005],c[500005],du[100005];
bool vis[100005];
int mi=0,id=0;
void dfs(int i)
{
	if(vis[i])return;
	vis[i]=1;
	if(c[a[i]]<mi)
	{
		mi=c[a[i]];
		id=a[i];
	}
	dfs(a[i]);
}
void solve()
{
	memset(vis,0,sizeof(vis));
	memset(du,0,sizeof(du));
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		du[a[i]]++;//记录入度 
	}
	for(int i=1;i<=n;i++)cin>>c[i];
	queue<int> q;
	vector<int> ans;
	//拓扑排序 （对非环上的点）
	for(int i=1;i<=n;i++)
	{
		if(!du[i])q.push(i);//入读为0的点入队 
	}
	while(!q.empty())
	{
		int k=q.front();
		q.pop();
		vis[k]=1;
		ans.push_back(k);
		du[a[k]]--;//删去一个点后，使它指向的点入度减一 
		if(!du[a[k]])q.push(a[k]);
	}
	//环上的点
	for(int i=1;i<=n;i++)
	{
		if(!vis[i])
		{
			mi=c[i];
			id=i;
			dfs(i);//找到价格最小的动物 
			int x=a[id];
			//卖掉环上的动物 
			do
			{
				ans.push_back(x);
				x=a[x];
			}while(x!=a[id]);
		}
	}
	for(int i=0;i<ans.size();i++)cout<<ans[i]<<" ";
	cout<<endl;
}
signed main()
{
	int t=1;
	cin>>t;
	while(t--)solve();
	return 0;
}