- 单向边森林不能通过dfs找环
- 抽象的题目需要通过模拟样例理解题意
- 列表模拟样例
- 链式前向星双倍空间(边&边的访问函数)
*
#include <bits/stdc++.h>
using namespace std;
const int mod=1000000007;
long long a[500005],b[500005],w[500005],c[500005],p[500005],cnt[500005];
long long h[500005],ne[1000005],t[1000005],tot,m,n;
bool v[500005],f[1000005];
long long jc[500005],jcinv[500005];
vector<long long>g[500005];
long long read1()
{
char cc=getchar();
while(!(cc>=48&&cc<=57))
{
if(cc=='-')
{
break;
}
cc=getchar();
}
bool f=false;
long long s=0;
if(cc=='-')
{
f=true;
}
else
{
s=cc-48;
}
while(1)
{
cc=getchar();
if(cc>=48&&cc<=57)
{
s=s*10+cc-48;
}
else
{
break;
}
}
if(f==true)
{
s=-s;
}
return s;
}
long long power(long long n,long long p)
{
if(p==0)
{
return 1;
}
else if(p==1)
{
return n%mod;
}
else if(p%2==0)
{
long long tmp=power(n,p/2);
return tmp*tmp%mod;
}
else
{
long long tmp=power(n,p/2);
return tmp*tmp%mod*n%mod;
}
}
void add(long long u,long long v)
{
tot++;
ne[tot]=h[u];
h[u]=tot;
t[tot]=v;
}
long long dfs1(long long n1)
{
v[n1]=true;
bool pd=false;
for(long long i=h[n1];i;i=ne[i])
{
if(f[i]==false)
{
f[i]=true;
if(i>n)
{
f[i-n]=true;
}
else
{
f[i+n]=true;
}
long long w=t[i];
if(v[w]==false)
{
long long tmp=dfs1(w);
if(tmp>0)
{
pd=true;
}
}
else
{
m++;
c[w]=m;
g[m].push_back(w);
pd=true;
}
}
}
if(pd==true)
{
if(c[n1]==0)
{
g[m].push_back(n1);
return c[n1]=m;
}
else
{
return 0;
}
}
return 0;
}
void dfs2(long long n1)
{
v[n1]=true;
for(long long i=h[n1];i;i=ne[i])
{
long long w=t[i];
if(i<=n&&v[w]==false)
{
if(p[w]==-1)
{
if(p[n1]==0)
{
p[w]=0;
cnt[w]=0;
}
else
{
cnt[w]=cnt[n1]+1;
p[w]=jcinv[cnt[w]];
}
}
dfs2(w);
}
}
}
void pre()
{
jc[0]=1;
jcinv[0]=1;
for(long long i=1;i<=500000;i++)
{
jc[i]=jc[i-1]*i%mod;
}
jcinv[500000]=power(jc[500000],1000000005);
for(long long i=500000;i>1;i--)
{
jcinv[i-1]=jcinv[i]*i%mod;
}
}
int main()
{
// freopen("H.in","r",stdin);
// freopen("H.out","w",stdout);
pre();
long long T;
cin>>T;
while(T--)
{
n=read1();
for(long long i=1;i<=n;i++)
{
a[i]=read1();
p[i]=-1;
cnt[i]=0;
}
for(long long i=1;i<=n;i++)
{
b[i]=read1();
add(b[i],i);
}
for(long long i=1;i<=n;i++)
{
add(i,b[i]);
}
for(long long i=1;i<=n;i++)
{
w[i]=read1();
}
for(long long i=1;i<=n;i++)
{
if(a[i]<a[b[i]])
{
p[i]=1;
cnt[i]=1;
}
else if(a[i]>=a[b[i]]+w[b[i]])
{
p[i]=0;
cnt[i]=0;
}
}
for(long long i=1;i<=n;i++)
{
if(!v[i])
{
long long tmp=dfs1(i);
}
}
for(long long i=1;i<=n;i++)
{
v[i]=false;
}
for(long long i=1;i<=m;i++)
{
long long x=-1;
for(long long j=0;j<g[i].size();j++)
{
if(p[g[i][j]]!=-1)
{
x=g[i][j];
}
}
if(x==-1)
{
for(long long j=0;j<g[i].size();j++)
{
p[g[i][j]]=0;
cnt[g[i][j]]=0;
}
x=g[i][0];
}
dfs2(x);
}
for(long long i=1;i<n;i++)
{
printf("%lld ",(a[i]+p[i]*w[i]%mod)%mod);
}
printf("%lld\n",(a[n]+p[n]*w[n]%mod)%mod);
for(long long i=1;i<=tot;i++)
{
f[i]=false;
}
tot=0;
for(long long i=1;i<=n;i++)
{
v[i]=false;
h[i]=0;
c[i]=0;
}
for(long long i=1;i<=m;i++)
{
g[i].clear();
}
m=0;
}
return 0;
}
标签:cc,Sweet,1000005,long,Sugar,II,n1,500005,true
From: https://www.cnblogs.com/watersail/p/18003755