QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#187065 | #3840. Pass the Ball! | bzw | WA | 4ms | 34520kb | C++14 | 2.3kb | 2023-09-24 14:14:15 | 2023-09-24 14:14:15 |
Judging History
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'