#283463 (C++14) No.732 3PrimeCounting

提出ソース

結果

問題	No.732 3PrimeCounting
ユーザー	beet
提出日時	2018-09-07 21:53:50
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	316 ms / 3,000 ms
コード長	2,764 bytes
コンパイル時間	1,875 ms
コンパイル使用メモリ	176,148 KB
実行使用メモリ	32,412 KB
最終ジャッジ日時	2024-11-29 12:59:07
合計ジャッジ時間	8,331 ms
ジャッジサーバーID （参考情報）	judge2 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,248 KB
testcase_02	AC	2 ms 5,248 KB
testcase_03	AC	2 ms 5,248 KB
testcase_04	AC	2 ms 5,248 KB
testcase_05	AC	2 ms 5,248 KB
testcase_06	AC	2 ms 5,248 KB
testcase_07	AC	2 ms 5,248 KB
testcase_08	AC	2 ms 5,248 KB
testcase_09	AC	2 ms 5,248 KB
testcase_10	AC	2 ms 5,248 KB
testcase_11	AC	2 ms 5,248 KB
testcase_12	AC	2 ms 5,248 KB
testcase_13	AC	2 ms 5,248 KB
testcase_14	AC	2 ms 5,248 KB
testcase_15	AC	2 ms 5,248 KB
testcase_16	AC	2 ms 5,248 KB
testcase_17	AC	2 ms 5,248 KB
testcase_18	AC	2 ms 5,248 KB
testcase_19	AC	2 ms 5,248 KB
testcase_20	AC	3 ms 5,248 KB
testcase_21	AC	4 ms 5,248 KB
testcase_22	AC	5 ms 5,248 KB
testcase_23	AC	2 ms 5,248 KB
testcase_24	AC	2 ms 5,248 KB
testcase_25	AC	8 ms 5,248 KB
testcase_26	AC	7 ms 5,248 KB
testcase_27	AC	3 ms 5,248 KB
testcase_28	AC	3 ms 5,248 KB
testcase_29	AC	5 ms 5,248 KB
testcase_30	AC	4 ms 5,248 KB
testcase_31	AC	5 ms 5,248 KB
testcase_32	AC	7 ms 5,248 KB
testcase_33	AC	7 ms 5,248 KB
testcase_34	AC	7 ms 5,248 KB
testcase_35	AC	6 ms 5,248 KB
testcase_36	AC	6 ms 5,248 KB
testcase_37	AC	3 ms 5,248 KB
testcase_38	AC	3 ms 5,248 KB
testcase_39	AC	7 ms 5,248 KB
testcase_40	AC	5 ms 5,248 KB
testcase_41	AC	5 ms 5,248 KB
testcase_42	AC	5 ms 5,248 KB
testcase_43	AC	4 ms 5,248 KB
testcase_44	AC	5 ms 5,248 KB
testcase_45	AC	3 ms 5,248 KB
testcase_46	AC	4 ms 5,248 KB
testcase_47	AC	4 ms 5,248 KB
testcase_48	AC	8 ms 5,248 KB
testcase_49	AC	9 ms 5,248 KB
testcase_50	AC	5 ms 5,248 KB
testcase_51	AC	5 ms 5,248 KB
testcase_52	AC	3 ms 5,248 KB
testcase_53	AC	17 ms 5,504 KB
testcase_54	AC	142 ms 18,476 KB
testcase_55	AC	140 ms 18,480 KB
testcase_56	AC	140 ms 18,480 KB
testcase_57	AC	37 ms 7,808 KB
testcase_58	AC	37 ms 7,680 KB
testcase_59	AC	17 ms 5,504 KB
testcase_60	AC	60 ms 10,716 KB
testcase_61	AC	60 ms 10,712 KB
testcase_62	AC	160 ms 19,108 KB
testcase_63	AC	83 ms 11,776 KB
testcase_64	AC	66 ms 11,152 KB
testcase_65	AC	67 ms 11,152 KB
testcase_66	AC	2 ms 5,248 KB
testcase_67	AC	2 ms 5,248 KB
testcase_68	AC	149 ms 18,812 KB
testcase_69	AC	148 ms 18,776 KB
testcase_70	AC	142 ms 18,504 KB
testcase_71	AC	142 ms 18,368 KB
testcase_72	AC	80 ms 11,648 KB
testcase_73	AC	212 ms 20,928 KB
testcase_74	AC	214 ms 20,864 KB
testcase_75	AC	8 ms 5,248 KB
testcase_76	AC	148 ms 18,644 KB
testcase_77	AC	39 ms 7,808 KB
testcase_78	AC	183 ms 19,840 KB
testcase_79	AC	151 ms 18,672 KB
testcase_80	AC	173 ms 19,584 KB
testcase_81	AC	141 ms 18,516 KB
testcase_82	AC	3 ms 5,248 KB
testcase_83	AC	38 ms 7,680 KB
testcase_84	AC	53 ms 10,452 KB
testcase_85	AC	131 ms 17,916 KB
testcase_86	AC	178 ms 19,712 KB
testcase_87	AC	316 ms 32,412 KB
testcase_88	AC	313 ms 32,412 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include<bits/stdc++.h>
using namespace std;
using Int = long long;

Int isprime(Int x){
  if(x<=2) return 0;
  for(Int i=2;i*i<=x;i++)
    if(x%i==0) return 0;
  return 1;
}


namespace FFT{
  using dbl = double;
  
  struct num{
    dbl x,y;
    num(){x=y=0;}
    num(dbl x,dbl y):x(x),y(y){}
  };
  
  inline num operator+(num a,num b){
    return num(a.x+b.x,a.y+b.y);
  }
  inline num operator-(num a,num b){
    return num(a.x-b.x,a.y-b.y);
  }
  inline num operator*(num a,num b){
    return num(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);
  }
  inline num conj(num a){
    return num(a.x,-a.y);
  }
 
  Int base=1;
  vector<num> rts={{0,0},{1,0}};
  vector<Int> rev={0,1};
 
  const dbl PI=acosl(-1.0);
 
  void ensure_base(Int nbase){
    if(nbase<=base) return;
 
    rev.resize(1<<nbase);
    for(Int i=0;i<(1<<nbase);i++)
      rev[i]=(rev[i>>1]>>1)+((i&1)<<(nbase-1));
 
    rts.resize(1<<nbase);
    while(base<nbase){
      dbl angle=2*PI/(1<<(base+1));
      for(Int i=1<<(base-1);i<(1<<base);i++){
	rts[i<<1]=rts[i];
	dbl angle_i=angle*(2*i+1-(1<<base));
	rts[(i<<1)+1]=num(cos(angle_i),sin(angle_i));
      }
      base++;
    }
  }
 
  void fft(vector<num> &a,Int n=-1){
    if(n==-1) n=a.size();
    assert((n&(n-1))==0);
 
    Int zeros=__builtin_ctz(n);
    ensure_base(zeros);
    Int shift=base-zeros;
    for(Int i=0;i<n;i++)
      if(i<(rev[i]>>shift))
	swap(a[i],a[rev[i]>>shift]);
 
    for(Int k=1;k<n;k<<=1){
      for(Int i=0;i<n;i+=2*k){
	for(Int j=0;j<k;j++){
	  num z=a[i+j+k]*rts[j+k];
	  a[i+j+k]=a[i+j]-z;
	  a[i+j]=a[i+j]+z;
	}
      }
    }
  }
 
  vector<num> fa;
 
  vector<Int> multiply(vector<Int> &a,vector<Int> &b){
    Int need=a.size()+b.size()-1;
    Int nbase=0;
    while((1<<nbase)<need) nbase++;
    ensure_base(nbase);
 
    Int sz=1<<nbase;
    if(sz>(Int)fa.size()) fa.resize(sz);
    for(Int i=0;i<sz;i++){
      Int x=(i<(Int)a.size()?a[i]:0);
      Int y=(i<(Int)b.size()?b[i]:0);
      fa[i]=num(x,y);
    }
    fft(fa,sz);
 
    num r(0,-0.25/sz);
    for(Int i=0;i<=(sz>>1);i++){
      Int j=(sz-i)&(sz-1);
      num z=(fa[j]*fa[j]-conj(fa[i]*fa[i]))*r;
      if(i!=j)
	fa[j]=(fa[i]*fa[i]-conj(fa[j]*fa[j]))*r;
      fa[i]=z;
    }
    fft(fa,sz);
 
    vector<Int> res(need);
    for(Int i=0;i<need;i++)
      res[i]=fa[i].x+0.5;
    
    return res;
  }
  
};

//INSERT ABOVE HERE
signed main(){
  Int n;
  cin>>n;

  vector<Int> v(n+1),w(2*n+1,0);
  for(Int i=0;i<=n;i++) v[i]=isprime(i);
  for(Int i=0;i<=n;i++) w[i*2]=isprime(i);
  
  auto x=FFT::multiply(v,v);
  auto y=FFT::multiply(v,x);
  auto z=FFT::multiply(v,w);

  Int ans=0;
  for(Int i=0;i<=3*n;i++){
    if(!isprime(i)) continue;
    ans+=y[i];
    ans-=z[i]*3;
    //cout<<i<<":"<<y[i]<<" "<<z[i]<<endl;
  }
  
  cout<<ans/6<<endl;
  return 0;
}

yukicoder

結果

テストケース

ソースコード