HDU6038(多校联合F题)

题意:
如果b中环的长度是a中环长度的因子,答案+(a[i]*j*b[j])
跑两遍dfs找出a中长度为i环的个数记为as[i],和b中长度为j环的个数bs[j]
然后跑循环计算就可以了.
code:

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int mod=1e9+7;
typedef long long ll;
const int maxn=1e5+10;
int a[maxn];
int b[maxn];
int as[maxn];
int bs[maxn];
bool vis[maxn];
int ans=1;
void dfs(int t,int l,int *a,int *b)
{
    if(vis[t])
    {
        b[l]++;
        return ;
    }
    vis[t]=1;
    dfs(a[t],l+1,a,b);
}
int main()
{
    int n;
    int m;
    int kase=0;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        for(int i=0;i<n;i++)
        {
            scanf("%d",&a[i]);
        }
        for(int i=0;i<m;i++)
        {
            scanf("%d",&b[i]);
        }
        memset(as,0,sizeof(as));
        memset(bs,0,sizeof(bs));
        memset(vis,0,sizeof(vis));
        //cout<<"-------------"<<endl;
        for(int i=0;i<m;i++)
        {
            if(!vis[i])
                dfs(i,0,b,bs);
        }
        memset(vis,0,sizeof(vis));
        for(int i=0;i<n;i++)
        {
            if(!vis[i])
                dfs(i,0,a,as);
        }
        ans=1;
        //cout<<"-------------"<<endl;
        for(int i=1;i<=n;i++)
        {
            if(as[i])
            {
                int res=0;
                for(int j=1;j*j<=i;j++)
                {
                    if(i%j==0)
                    {
                        (res+=(ll)bs[j]%mod*j%mod)%=mod;
                        if(j*j!=i)
                            (res+=(ll)bs[i/j]%mod*(i/j)%mod)%=mod;
                    }
                }
                for(int j=1;j<=as[i];j++)
                    ans=(ll)ans *res%mod;
            }
        }
        //cout<<"-------------"<<endl;
        printf("Case #%d: %d\n", ++kase, ans);
    }
}