Codeforces 1100 E. Andrew and Taxi 二分+拓扑排序

Solution

显然答案具有二分性……check时$\le mid$的边是可以当做不存在的。
问题是如何输出方案。
先说正解，弄出二分时最后一次合法的拓扑序，对于一条边$x->y$，如果拓扑序与之矛盾这条边就要变方向。
我的方法则比较麻烦，对于每个点，搞两个入度，一个是原图入度，一个是可以变方向的边产生的入度，每次找这两个入度相同的点，把连向它的边全都变方向，正确性显然，但代码长不少。

Code

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=100010;
const int inf=2147483647;
int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
	return x*f;
}
int n,m,td[Maxn];
struct Edge{int x,y,d,next,id;}e[Maxn],E[Maxn];
bool cmp(Edge a,Edge b){return a.d<b.d;}
int len=0,last[Maxn];
void ins(int x,int y,int d)
{
	int t=++len;
	e[t].x=x;e[t].y=y;e[t].d=d;e[t].next=last[x];last[x]=t;
}
bool check(int d)
{
	queue<int>q;int cnt=0;
	len=0;memset(last,0,sizeof(last));
	memset(td,0,sizeof(td));
	for(int i=1;i<=m;i++)if(E[i].d>d)ins(E[i].x,E[i].y,E[i].d),td[E[i].y]++;
	for(int i=1;i<=n;i++)if(!td[i])q.push(i);
	while(!q.empty())
	{
		int x=q.front();q.pop();cnt++;
		for(int i=last[x];i;i=e[i].next)
		{
			int y=e[i].y;
			td[y]--;
			if(!td[y])q.push(y);
		}
	}
	return(cnt==n);
}
vector<int>ans,edge[Maxn];bool mark[Maxn],del[Maxn],in[Maxn];
int dd[Maxn],deg[Maxn];
queue<int>Q;
int main()
{
	n=read(),m=read();int l=0,r=-inf;
	for(int i=1;i<=m;i++)
	{
		int x=read(),y=read(),d=read();deg[y]++;
		E[i].x=x,E[i].y=y,E[i].d=d,E[i].id=i;r=max(r,d);
	}
	sort(E+1,E+1+m,cmp);
	while(l<=r)
	{
		int mid=l+r>>1;
		if(check(mid))r=mid-1;
		else l=mid+1;
	}
	printf("%d ",r+1);
	len=0;memset(last,0,sizeof(last));
	for(int i=1;i<=m;i++)ins(E[i].x,E[i].y,E[i].d);
	queue<int>q;
	for(int i=1;i<=n;i++)if(!deg[i])q.push(i);
	while(!q.empty())
	{
		int x=q.front();q.pop();mark[x]=true;
		for(int i=last[x];i;i=e[i].next)
		{
			int y=e[i].y;
			deg[y]--;
			if(!deg[y])q.push(y);
		}
	}
	bool all=true;
	for(int x=1;x<=n;x++)if(!mark[x]){all=false;break;}
	if(all){puts("0");return 0;}
	int rest=0;
	for(int x=1;x<=n;x++)
	if(!mark[x])
	{
		rest++;
		for(int i=last[x];i;i=e[i].next)
		{
			int y=e[i].y;
			if(e[i].d<=r+1)dd[y]++,edge[y].push_back(i);
		}
	}
	for(int x=1;x<=n;x++)
	if(!mark[x]&&deg[x]==dd[x])Q.push(x);
	while(rest)
	{
		while(!Q.empty())
		{
			int x=Q.front();Q.pop();
			if(mark[x]||in[x])continue;
			in[x]=true;q.push(x);
		}
		while(!q.empty())
		{
			int x=q.front();q.pop();mark[x]=true;rest--;
			for(int i=0;i<edge[x].size();i++)
			{
				int t=edge[x][i];
				if(del[t])continue;
				ans.push_back(E[t].id);del[t]=true;
			}
			for(int i=last[x];i;i=e[i].next)
			{
				if(del[i])continue;
				int y=e[i].y;
				del[i]=true;deg[y]--;
				if(e[i].d<=r+1)dd[y]--;
				if(!deg[y]&&!in[y])q.push(y);
				if(deg[y]==dd[y])Q.push(y);
			}
		}
	}
	printf("%d\n",ans.size());
	sort(ans.begin(),ans.end());
	for(int i=0;i<ans.size();i++)printf("%d ",ans[i]);
}