hdu 4696 Answers/杭电2013年多校第十场循环

最新推荐文章于 2024-07-27 19:05:14 发布

knownothing

最新推荐文章于 2024-07-27 19:05:14 发布

阅读量1k

点赞数

分类专栏：数论* 文章标签：循环

本文链接：https://blog.csdn.net/a601025382s/article/details/10190311

版权

数论* 专栏收录该内容

113 篇文章 0 订阅

订阅专栏

#include <cstdio>
#include <cstring>
#include <cmath>
#include <vector>
#include <map>
#include <algorithm>
#include <iostream>
using namespace std;
#define LL __int64
#define maxn 100011
int t[maxn],c[maxn],n,q,m[maxn],vis[maxn],a[maxn],b[maxn];
map<int,map<int,int> >nn;
map<int,int>mm;
map<int,int>ll;
int bb[1001];
int flag[maxn],s[maxn],ss[maxn];
void find(int x)
{
    ll.clear();
    int tt=1,i,j,k,y,z,d;
    s[0]=0;s[1]=c[x];ll[x]=1,ss[1]=x;
    for(i=2;;i++)
    {
        vis[x]=1;
        x=t[x];
        ss[i]=x;
        if(ll[x])
            break;
        if(vis[x])
        {
            for(j=i-1;j>=1;j--)
            {
                a[ss[j]]=s[i-1]-s[j-1]+a[x];
                b[ss[j]]=b[x];
            }
            return;
        }
        ll[x]=++tt;
        s[i]=s[i-1]+c[x];
        //cout<<s[i]<<endl;
    }
    d=s[i-1]-s[ll[x]-1];
    for(j=1;j<i;j++)
    {
        if(j<=ll[x])
            a[ss[j]]=s[ll[x]-1]-s[j-1];
        else
            a[ss[j]]=s[i-1]-s[j-1];
        b[ss[j]]=d;
    }
}
int main()
{
    while(scanf("%d%d",&n,&q)!=EOF)
    {
        int i,j,k;
        for(i=1;i<=n;i++)
            scanf("%d",&t[i]);
        for(i=1;i<=n;i++)
            scanf("%d",&c[i]);
        for(i=1;i<=q;i++)
            scanf("%d",&m[i]);
        memset(vis,0,sizeof(vis));
        for(i=1;i<=n;i++)
        {
            if(!vis[i])
            find(i);
        }
       // for(i=1;i<=n;i++)
        //    cout<<i<<" "<<a[i]<<" "<<b[i]<<endl;
        mm.clear();
        nn.clear();
        int tt=0;
        for(i=1;i<=n;i++)
        {
            if(mm[b[i]]==0)
            {
                mm[b[i]]=++tt;
                bb[tt]=b[i];
            }
            nn[tt][a[i]]=1;
        }
        //cout<<nn[1][0]<<endl;
        memset(flag,0,sizeof(flag));
        for(j=1;j<=q;j++)
        {
            if(m[j]<=0)continue;
            for(i=1;i<=tt;i++)
            if(nn[i][m[j]%bb[i]]){flag[j]=1;break;}
        }
        for(i=1;i<=q;i++)
            if(flag[i])printf("YES\n");
            else printf("NO\n");
    }
    return 0;
}
/*
    由于t[i]<=n,t(xi-1)=xi.所以必定会出现循环。枚举x1=i(1<=i<=n)可求出a[i],b[i]
    a[i],b[i]表示，在经过a[i]长度后会进入b[i]长度的循环，所以符合a[i]+b[i]*k==m[i]的m[i]都是YES
    直接暴力的话就是对于每个m[i]逐次匹配是否能在a[i],b[i]的循环中出现。最坏情况需要O(n^2)，会超时；
    我们可以发现有很多数的b[i]是相同的，所以我们可以枚举b[i]减少时间(在一个循环中出现的数b[i]肯定相同)
    
    由于t(xi-1)=xi，所以在枚举的i中循环出现过j，
    a[j]=a[i]-s[j'-1](s[i]表示在枚举过程中前i个数的c[i]和,j'表示j所在位置),b[i]=b[j];
    所以可以再O(n)时间内，找出所有a[i],b[i]
*/