2021“MINIEYE杯”中国大学生算法设计超级联赛（7） Link with Limit(tarjan)

Link with Limit

题目大意：

给一个数组$a[]$，求

$g(x)$是否收敛，乍看是数学题，其实是图论

每一个数据都是相关的，因此一定会形成形成图，且图中每一节点出度为$1$，所以我们的问题变成了，求图中入度不为$0$的一些点，也就是找这些图中的环中节点和的平均数是否相等（注意，是平均数）

解法：

我们可以反向建图，然后缩点，查看入度为0的点，就是我们要找到，也就是说缩点后，比较平均数即可

#include <bits/stdc++.h>
#define inf 0x7fffffff
//#define ll long long
#define int long long
//#define double long double
//#define double long long
#define re int
//#define void inline void
#define eps 1e-8
//#define mod 1e9+7
#define ls(p) p<<1
#define rs(p) p<<1|1
#define pi acos(-1.0)
#define pb push_back
#define mk make_pair
#define P pair < int , int >
using namespace std;
const int mod=4933;
//const int inf=1e18;
const int M=1e8;
const int N=4e6+5;//??????.???? 4e83
int n;
struct node
{
    int ver,next;
}e[N];
int tot,head[N];
int dfn[N],low[N],ins[N];
int instack[N],c[N];
int top,cnt,num;
int w[N],d[N],in[N],v[N];
void init()
{
    for(re i=1;i<=n;i++)  v[i]=in[i]=w[i]=d[i]=head[i]=c[i]=instack[i]=ins[i]=dfn[i]=low[i]=0;
    num=top=cnt=tot=0;
}
void add(int x,int y)
{
    e[++tot].ver=y;
    e[tot].next=head[x];
    head[x]=tot;
}
void tarjan(int x)
{
    dfn[x]=low[x]=++num;
    ins[x]=1;
    instack[++top]=x;
    for(re i=head[x];i;i=e[i].next)
    {
        int y=e[i].ver;
        if(!dfn[y])
        {
            tarjan(y);
            low[x]=min(low[x],low[y]);
        }
        else if(ins[y])  low[x]=min(low[x],dfn[y]);
    }
    if(low[x]==dfn[x])
    {
        int y;
        cnt++;
        do
        {
            y=instack[top--];
            c[y]=cnt;
            ins[y]=0;
        }while(x!=y);
    }
}
void solve()
{
    int op=0;
    vector < int > a;
    cin>>n;
    init();
    for(re i=1;i<=n;i++)
    {
        int x;
        scanf("%lld",&x);
        add(x,i);
    }
    for(re i=1;i<=n;i++)  if(!dfn[i])  tarjan(i);
    for(re i=1;i<=n;i++)
    {
        w[c[i]]+=i;
        d[c[i]]++;
    } 
    for(re x=1;x<=n;x++)  for(re i=head[x];i;i=e[i].next)
    {
        int y=e[i].ver;
        if(c[x]==c[y])  continue;
        in[c[y]]++;
    }
    for(re i=1;i<=cnt;i++)  if(!in[i])  a.pb(i);
    int sz=a.size();
    for(re i=0;i<sz;i++)  if(w[a[0]]*d[a[i]]!=w[a[i]]*d[a[0]])  op=1;
    puts(op?"NO":"YES");
}
signed main()
{
    int T=1;      
    cin>>T;
    for(int index=1;index<=T;index++)
    {
        solve();
//        puts("");
    }
    return 0;
}
/*
1
5
2 1
2 5
5 3
2 4
2 1
1 5
1 3
3 4


6
5 5 5 6 15
5 9 5 6 23


*/