QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#187065#3840. Pass the Ball!bzwWA 4ms34520kbC++142.3kb2023-09-24 14:14:152023-09-24 14:14:15

Judging History

This is the latest submission verdict.

  • [2023-09-24 14:14:15]
  • Judged
  • Verdict: WA
  • Time: 4ms
  • Memory: 34520kb
  • [2023-09-24 14:14:15]
  • Submitted

answer

// qwq
#include <bits/stdc++.h>
using namespace std;
constexpr int N=1e6+1;
struct cp{
    long double a,b;
    cp operator + (cp o){return {a+o.a,b+o.b};}
    cp operator - (cp o){return {a-o.a,b-o.b};}
    cp operator * (cp o){return {a*o.a-b*o.b,a*o.b+b*o.a};}
};
const long double pi=acos(-1);
cp w[N]; int rev[N];
void fft(cp* a,int len,bool ok){
    for(int i=0;i<len;i++){
        rev[i]=(rev[i>>1]>>1)+(i&1?len>>1:0);
        if(rev[i]>i)swap(a[rev[i]],a[i]);
    }
    for(int d=1;d<len;d<<=1){
        cp W={cos(pi/d),sin(pi/d)};if(ok)W.b=-W.b;
        w[0]={1.,0.}; for(int i=1;i<d;i++)w[i]=w[i-1]*W;
        for(int fi=0;fi<len;fi+=d<<1){
            int se=fi+d;
            for(int i=0;i<d;i++){
                cp a0=a[fi+i],a1=a[se+i]*w[i];
                a[fi+i]=(a0+a1),a[se+i]=(a0-a1);
            }
        }
    }
    if(ok){
        for(int i=0;i<len;i++)
            a[i].a/=len;
    }
}
typedef long long ll;
typedef vector<ll> arr;
cp f[N],g[N];
arr mul(arr x,arr y){
    int n=x.size(),m=y.size(),len=1;
    while(len<=(n*3+3))len<<=1;
    for(int i=0;i<len;i++)f[i]=g[i]={0.,0.};
    for(int i=0;i<n;i++)f[i+n].a=f[i].a=x[i];
    for(int i=0;i<n;i++)g[i].a=y[n-1-i];
    fft(f,len,0),fft(g,len,0);
    for(int i=0;i<len;i++)f[i]=f[i]*g[i];
    fft(f,len,1);
    arr cur(n,0);
    for(int i=0;i<n;i++)
        cur[i]=ll(f[n-1+i].a+0.5);
    return cur;
}
arr Mul(arr x,arr y){
    arr a=x,b=x,c=y,d=y;
    int n=x.size();
    for(int i=0;i<n;i++)
        a[i]/=100,b[i]%=100,
        c[i]/=100,d[i]%=100;
    arr E[4]={mul(a,b),mul(a,c),mul(b,d),mul(c,d)};
    for(int i=0;i<n;i++)
        a[i]=E[0][i]*10000ll+(E[1][i]+E[2][i])*100ll+E[3][i];
    return a;
}
arr e[N],s; int n,m,vis[N],p[N];
int main(){
    cin>>n>>m;for(int i=1;i<=n;i++)cin>>p[i];
    for(int i=1;i<=n;i++)if(!vis[i]){
        int x=i,t=0;arr a;
        while(!vis[x])a.push_back(x),vis[x]=1,x=p[x],t++;
        a=Mul(a,a);
        if(e[t].size()==0){
            s.push_back(t);
            e[t]=a;
        }
        else{
            for(int j=0;j<t;j++)
                e[t][j]+=a[j];
        }
    }
    while(m--){
        int q;cin>>q;
        ll ans=0;
        for(ll i:s)
            ans+=e[i][q%i];
        cout<<ans<<'\n';
    }
    return 0;
}

詳細信息

Test #1:

score: 0
Wrong Answer
time: 4ms
memory: 34520kb

input:

4 4
2 4 1 3
1
2
3
4

output:

2500
2000
2500
3000

result:

wrong answer 1st lines differ - expected: '25', found: '2500'