满脑子都是刚学的 Tarjan ,看题的时候甚至不知道怎么操作特判
由题意可知,控制所有的间谍一定需要满足:所有入度为 0 的间谍都能够收买。而对于其他入度不为
0
0
0 的点,在满足了以上的条件之后,是一定可以通过入度为
0
0
0 的点到达的 (这不是显然的吗)
因此输出 N O NO NO 的情况仅为:存在入度为 0 0 0 的间谍不能够收买。由于该情况下还需要输出编号最小且不能控制的间谍,先缩点再处理会比较麻烦,考虑直接把所有能收买的放进 t a r j a n tarjan tarjan 中跑一遍后,第一个未更新 d f n dfn dfn 值的即为所求。
确定有解后,对原图进行缩点操作,并在弹栈的过程中维护当前强连通分量中需要花费的最小值(即该强连通分量中能收买的间谍需要的最小花费),并根据原图统计新图的入度。最后的答案即为所有入度为 0 0 0 的强连通分量的花费之和。
#include <bits/stdc++.h>
#define int long long
#define PII pair<int,int>
using namespace std;
const int N=5e3+10;
vector <int> h[N];
bool vis[N],in[N];
int dfn[N],low[N],suo[N],num[N],a[N],value[N],money[N];
stack<int> st;
vector<PII> q;
int tmp=0,cnt=0,x=0;
int ans=0;
void tarjan(int u){
dfn[u]=low[u]=++x;
in[u]=true;
st.push(u);
for(auto v:h[u]){
if(!dfn[v]){
tarjan(v);
low[u]=min(low[u],low[v]);
}
else if(in[v])
low[u]=min(low[u],dfn[v]);
}
if(dfn[u]==low[u]){
cnt++;
do{
tmp=st.top();
st.pop();
in[tmp]=false;
suo[tmp]=cnt;
if(value[tmp]!=-1)
money[cnt]=min(money[cnt],value[tmp]);
}
while(u!=tmp);
}
}
signed main(){
int n,p,r,x,y,u,v;
for(int i=1;i<=3000;i++){
value[i]=-1;
money[i]=20005;
}
cin>>n>>p;
for(int i=1;i<=p;i++){
cin>>x>>y;
value[x]=y;
}
cin>>r;
for(int i=1;i<=r;i++){
cin>>u>>v;
q.push_back({u,v});
h[u].push_back(v);
}
for(int i=1;i<=n;i++){
if(value[i]!=-1&&!dfn[i])
tarjan(i);
}
for(int i=1;i<=n;i++){
if(!dfn[i]){
cout<<"NO"<<endl<<i;
return 0;
}
}
for(auto i:q){
if(suo[i.first]==suo[i.second])
continue;
num[suo[i.second]]++;
}
int sum=0;
for(int i=1;i<=cnt;i++){
if(num[i]==0&&!vis[i])
sum+=money[i],vis[i]=1;
}
cout<<"YES"<<endl<<sum;
return 0;
}