tarjan 算法求强连通分量

 #include<bits/stdc++.h>
 #define ll long long
 using namespace std;
 const int P=1e6;
 const int N=2e6+;
 const int M=2e6+;
 const int inf=0x3f3f3f3f;
 struct edge
 {
     int f,t,next;
     ll v;
 }e[M];
 int n,m,tot,all,s,cnt,head[N],dfn[N],low[N],st[N],id[N];
 int dindex;
 bool inst[N];
 ll num[N],sum[],val[N];
 void add(int f,int t,ll v)
 {
     e[tot].f=f; e[tot].t=t; e[tot].v=v;
     e[tot].next=head[f]; head[f]=tot++;
 }
 void tarjan(int v)
 {
     dindex++;
     dfn[v]=low[v]=dindex;
     st[all++]=v; inst[v]=true;
     for(int i=head[v];~i;i=e[i].next)
     {
         int nx=e[i].t;
         if(!dfn[nx])
         {
             tarjan(nx);
             low[v]=min(low[v],low[nx]);
         }
         else if(inst[nx]) low[v]=min(low[v],dfn[nx]);
     }
     if(dfn[v]==low[v])
     {
         cnt++;
         while()
         {
             int cur=st[--all];
             inst[cur]=false;
             id[cur]=cnt;
             if(cur==v) break;
         }
     }
 }
 ll work(ll x)
 {
     ll ans=;
     int item=lower_bound(num,num+,x)-num;
     ans-=sum[item];
     ans+=x*(item+);
     ans+=num[item]-x;
     return ans;
 }
 ll dfs(int v)
 {
     ll res=;
     dfn[v]=;
     for(int i=head[v];~i;i=e[i].next)
     {
         int nx=e[i].t;
         if(!dfn[nx]) res=max(res,e[i].v+dfs(nx));
         else res=max(res,val[nx]+e[i].v);
     }
     val[v]+=res;
     return val[v];
 }
 int main()
 {
     num[]=; cnt=; all=; tot=; dindex=;
     for(int i=;i<=;i++) num[i]=num[i-]+i,sum[i]=sum[i-]+num[i];
     memset(head,-,sizeof(head));
     scanf("%d%d",&n,&m);
     for(int i=;i<=m;i++)
     {
         int f,t; ll v;
         scanf("%d%d%lld",&f,&t,&v);
         add(f,t,v);
     }
     scanf("%d",&s);
     tarjan(s);
     int up=tot;
     for(int i=;i<up;i++)
     {
         int a=e[i].f,b=e[i].t;
         if(!id[a] || !id[b]) continue;
         if(id[a]==id[b]) val[id[a]+P]+=work(e[i].v);
         else if(id[a]<id[b]) add(id[b]+P,id[a]+P,e[i].v);
         else add(id[a]+P,id[b]+P,e[i].v);
     }
     ll ans=dfs(id[s]+P);
     printf("%lld\n",ans);
     return ;
 }