P3119 [USACO15JAN] Grass Cownoisseur G

题目传送门

P3119 [USACO15JAN] Grass Cownoisseur G

题意

一个有向图，每个点上有一个宝藏，走到这个点时可以获得这个宝藏，但只能获取一次。有一次走反向边的机会，从 $1$ 号点出发，最后回到 $1$ 号点。求最多获得的宝藏数。

思路

第一眼想到缩点。

因为强连通分量两点之间可以互相走到，这样可以保证获得宝藏数多且不需要使用机会。

先进行缩点，这样可以保证现在的图上无环。缩点后，我们考虑存图，可以用 $vector$ 或链式前向星。

建图后就是一个有向无环图 $(DAG)$。这里可以使用 $拓扑+dp$ 去进行其余的操作。当然，好像可以用 $分层图+spfa$ 水过，我这篇题解用的就是后者的方法。

考虑建两层图。除了原边，还要建跨两层的边。

对于每一条边，设这条边起始点为 $u$，终点为 $v$。则我们就将第一层的 $v$ 连向第二层的 $u$，可以达到走反向边的效果，剩下的边按照原图建即可。

最后跑一下最长路记录答案，输出即可。

代码

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
inline int read(){
    int x=0,f=1;char c=getchar();
    for(;!isdigit(c);c=getchar()) if(c=='-') f=-1;
    for(;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+c-'0';
    return x*f;
}
inline void write(int x){
    if(x<0) putchar('-'),x=-x;
    if(x>9) write(x/10);
    putchar(x%10+'0');
}
const int maxn=1e5+5;
int n,m,tot,cnt;
int dfn[maxn],low[maxn],siz[maxn<<1],dis[maxn<<1],bcc[maxn];
bool f[maxn<<1];
vector<int> g[maxn],g1[maxn<<1];
stack<int> st;
queue<int> q;
void dfs(int u){
	dfn[u]=low[u]=++tot,f[u]=1,st.push(u);
	for(auto v:g[u]){
		if(!dfn[v]) dfs(v),low[u]=min(low[u],low[v]);
		else if(f[v]) low[u]=min(low[u],dfn[v]);
	}
	if(dfn[u]==low[u]){
		cnt++;
		while(st.top()!=u) bcc[st.top()]=cnt,f[st.top()]=0,siz[cnt]++,st.pop();
		bcc[st.top()]=cnt,f[st.top()]=0,siz[cnt]++,st.pop();
	}
}
void spfa(){
	f[bcc[1]]=1,q.push(bcc[1]);
	while(!q.empty()){
		int x=q.front();
		for(auto v:g1[x]){
			if(dis[v]<dis[x]+siz[x]){
				dis[v]=dis[x]+siz[x];
				if(!f[v]) f[v]=1,q.push(v);
			}
		}
		q.pop(),f[x]=0;
	}
}
signed main(){
	n=read(),m=read();
	for(int i=1,u,v;i<=m;i++) u=read(),v=read(),g[u].push_back(v);
	for(int i=1;i<=n;i++) if(!dfn[i]) dfs(i);
	if(cnt==1) return cout<<n<<endl,0;
	for(int i=1;i<=cnt;i++) siz[cnt+i]=siz[i];
	for(int i=1;i<=n;i++) for(auto v:g[i]) if(bcc[i]!=bcc[v]) g1[bcc[i]].push_back(bcc[v]),g1[bcc[v]].push_back(bcc[i]+cnt),g1[bcc[i]+cnt].push_back(bcc[v]+cnt);
	spfa();
	cout<<dis[bcc[1]+cnt]<<endl;
	return 0;
}

提交记录

注意：当整个图为强连通分量时，直接输出 $n$ 即可。