缩点模板

题目:洛谷3389

缩点本质上就是把一些可以相互到达(强连通分量)的点看成一个点,然后构建一个新图,方便计算。

#include
using namespace std;
const int N=10005;
const int M=100005;
int n,m,head[N],dfn[N],low[N],vis[N],s[N],top,cnt,ans;
int Head[N],w[N],D[N],F[N],tot,in[N];//D[]:DP数组,F[]:记录那几个点属于同一个强连通分量,in[]:入度 
struct node{
	int x,y,next;
}e[M],ne[M];//旧图 / 新图 
void Tarjan(int u){
	dfn[u]=low[u]=++cnt,vis[u]=1,s[++top]=u;
	for(int i=head[u],v;v=e[i].y,i;i=e[i].next)//Tarjan 
		if(!dfn[v]){
			Tarjan(v);
			low[u]=min(low[u],low[v]);
		}
		else if(vis[v]) low[u]=min(low[u],dfn[v]);
	if(low[u]==dfn[u])//合并 
		while(int k=s[top--]){
			vis[k]=0,F[k]=u;//记录从属关系 
			if(k==u) break;
			w[u]+=w[k];//点权合并 
		}
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>w[i];
	for(int i=1,x,y;i<=m;i++){
		cin>>x>>y;
		e[i].x=x,e[i].y=y,e[i].next=head[x],head[x]=i;
	}
	for(int i=1;i<=n;i++)
		if(!dfn[i]) Tarjan(i);
	for(int i=1,x,y;x=F[e[i].x],y=F[e[i].y],i<=m;i++)
		if(x!=y)//如果x,y不属于同一个强连通分量 
			ne[++tot].x=x,ne[tot].y=y,ne[tot].next=Head[x],Head[x]=tot,in[y]++;//建立新图 
	top=0;
	for(int i=1;i<=n;i++)
		if(F[i]==i&&!in[i]) s[++top]=i,D[i]=w[i];//入度为0即可作为起点 
	while(top){
		int k=s[top--];
		for(int i=Head[k],v;v=ne[i].y,i;i=ne[i].next){//DP
			D[v]=max(D[v],D[k]+w[v]);
			if(--in[v]==0) s[++top]=v;
		}
	}
	for(int i=1;i<=n;i++) ans=max(ans,D[i]);
	cout<

你可能感兴趣的:(模板,图论)