洛谷3387 模板 缩点

题目：缩点

思路：tarjan缩点+最长路。

注意：
1、用dijkstra求最长路时，优先队列中的<运算符要反过来。
2、需要把所有入度为0的点为起点跑一遍最长路。
3、每次求最长路时，dist数组不要重新初始化。
4、缩点后的点权为缩点前的所有点权之和。
5、最开始初始化dist[s]的值不为0，为点权。

代码：#include<bits/stdc++.h>
using namespace std;

#define maxn 10000
#define maxm 100000

struct Pair {
int x;
int y;
Pair(int xx=0,int yy=0) {
x=xx,y=yy;
}
bool operator < (const Pair& oth) const {
return x<oth.x||(x==oth.x&&y<oth.y);
}
};

int n,m;
int w[maxn+5];
vector<int> a[maxn+5];

int pre[maxn+5]= {0},low[maxn+5],clk=0;
stack<int> s;
int cnt=0,isin[maxn+5];

vector<int> g[maxn+5];
int goin[maxn+5]= {0};
int dist[maxn+5]= {0};
int val[maxn+5]={0};

void readin() {
scanf("%d%d",&n,&m);
for(int i=1; i<=n; i++) {
scanf("%d",&w[i]);
}
for(int i=1; i<=m; i++) {
int x,y;
scanf("%d%d",&x,&y);
a[x].push_back(y);
}
}

void tarjan(int x) {
pre[x]=low[x]=++clk;
s.push(x);
for(int i=0; i<a[x].size(); i++) {
int y=a[x][i];
if(!pre[y]) {
tarjan(y);
low[x]=min(low[x],low[y]);
} else if(!isin[y]) {
low[x]=min(low[x],pre[y]);
}
}
if(pre[x]==low[x]) {
cnt++;
int u;
do {
u=s.top();
s.pop();
isin[u]=cnt;
val[cnt]+=w[u];
} while(u!=x);
}
}

void make_g() {
for(int i=1;i<=n;i++){
for(int j=0;j<a[i].size();j++){
if(isin[i]!=isin[a[i][j]]) {
g[isin[i]].push_back(isin[a[i][j]]);
goin[isin[a[i][j]]]++;
}
}
}
}

void dijkstra(int st) {
dist[st]=val[st];
priority_queue<Pair> p;
p.push(Pair(dist[st],st));
bool c[maxn+5]= {0};
while(!p.empty()) {
Pair x=p.top();
p.pop();
if(c[x.y]) continue;
c[x.y]=true;
for(int i=0; i<g[x.y].size(); i++) {
if(dist[g[x.y][i]]<x.x+val[g[x.y][i]]) {
dist[g[x.y][i]]=x.x+val[g[x.y][i]];
p.push(Pair(dist[g[x.y][i]],g[x.y][i]));
}
}
}
}

int main() {
readin();
for(int i=1; i<=n; i++) if(!pre[i]) tarjan(i);
make_g();
for(int i=1; i<=cnt; i++) {
if(!goin[i]) dijkstra(i);
}
int ans=0;
for(int i=1;i<=cnt;i++){
ans=max(dist[i],ans);
}
printf("%d",ans);
return 0;
}