結果

問題 No.732 3PrimeCounting
ユーザー beetbeet
提出日時 2018-09-08 00:02:31
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 134 ms / 3,000 ms
コード長 2,845 bytes
コンパイル時間 1,784 ms
コンパイル使用メモリ 176,960 KB
実行使用メモリ 34,892 KB
最終ジャッジ日時 2024-05-07 10:00:00
合計ジャッジ時間 5,951 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 7 ms
5,888 KB
testcase_01 AC 6 ms
6,016 KB
testcase_02 AC 6 ms
6,016 KB
testcase_03 AC 6 ms
6,016 KB
testcase_04 AC 6 ms
6,144 KB
testcase_05 AC 6 ms
6,016 KB
testcase_06 AC 6 ms
6,144 KB
testcase_07 AC 6 ms
6,144 KB
testcase_08 AC 7 ms
6,016 KB
testcase_09 AC 6 ms
6,144 KB
testcase_10 AC 6 ms
6,016 KB
testcase_11 AC 6 ms
6,016 KB
testcase_12 AC 6 ms
6,016 KB
testcase_13 AC 7 ms
6,016 KB
testcase_14 AC 6 ms
5,888 KB
testcase_15 AC 6 ms
6,016 KB
testcase_16 AC 6 ms
6,016 KB
testcase_17 AC 6 ms
6,016 KB
testcase_18 AC 6 ms
6,144 KB
testcase_19 AC 6 ms
6,144 KB
testcase_20 AC 6 ms
6,144 KB
testcase_21 AC 8 ms
6,528 KB
testcase_22 AC 8 ms
6,400 KB
testcase_23 AC 6 ms
6,144 KB
testcase_24 AC 7 ms
6,016 KB
testcase_25 AC 10 ms
6,912 KB
testcase_26 AC 9 ms
6,912 KB
testcase_27 AC 7 ms
6,392 KB
testcase_28 AC 6 ms
6,144 KB
testcase_29 AC 8 ms
6,400 KB
testcase_30 AC 8 ms
6,656 KB
testcase_31 AC 8 ms
6,400 KB
testcase_32 AC 10 ms
7,040 KB
testcase_33 AC 9 ms
6,940 KB
testcase_34 AC 9 ms
7,040 KB
testcase_35 AC 9 ms
6,912 KB
testcase_36 AC 9 ms
6,940 KB
testcase_37 AC 6 ms
6,016 KB
testcase_38 AC 6 ms
6,144 KB
testcase_39 AC 9 ms
6,912 KB
testcase_40 AC 8 ms
6,528 KB
testcase_41 AC 8 ms
6,528 KB
testcase_42 AC 7 ms
6,400 KB
testcase_43 AC 8 ms
6,272 KB
testcase_44 AC 8 ms
6,276 KB
testcase_45 AC 6 ms
6,272 KB
testcase_46 AC 7 ms
6,272 KB
testcase_47 AC 8 ms
6,400 KB
testcase_48 AC 10 ms
7,040 KB
testcase_49 AC 10 ms
6,912 KB
testcase_50 AC 8 ms
6,272 KB
testcase_51 AC 8 ms
6,400 KB
testcase_52 AC 7 ms
6,144 KB
testcase_53 AC 13 ms
7,936 KB
testcase_54 AC 61 ms
20,824 KB
testcase_55 AC 63 ms
20,828 KB
testcase_56 AC 62 ms
20,832 KB
testcase_57 AC 21 ms
10,112 KB
testcase_58 AC 21 ms
10,112 KB
testcase_59 AC 14 ms
8,064 KB
testcase_60 AC 34 ms
13,188 KB
testcase_61 AC 33 ms
13,032 KB
testcase_62 AC 64 ms
21,584 KB
testcase_63 AC 37 ms
14,336 KB
testcase_64 AC 34 ms
13,628 KB
testcase_65 AC 35 ms
13,628 KB
testcase_66 AC 6 ms
6,016 KB
testcase_67 AC 6 ms
6,016 KB
testcase_68 AC 64 ms
21,256 KB
testcase_69 AC 65 ms
21,128 KB
testcase_70 AC 63 ms
20,852 KB
testcase_71 AC 62 ms
20,852 KB
testcase_72 AC 37 ms
14,336 KB
testcase_73 AC 71 ms
23,296 KB
testcase_74 AC 71 ms
23,424 KB
testcase_75 AC 9 ms
6,944 KB
testcase_76 AC 64 ms
21,088 KB
testcase_77 AC 21 ms
10,240 KB
testcase_78 AC 67 ms
22,400 KB
testcase_79 AC 63 ms
21,144 KB
testcase_80 AC 69 ms
21,760 KB
testcase_81 AC 62 ms
20,872 KB
testcase_82 AC 7 ms
6,260 KB
testcase_83 AC 21 ms
10,240 KB
testcase_84 AC 34 ms
12,924 KB
testcase_85 AC 63 ms
20,392 KB
testcase_86 AC 69 ms
22,016 KB
testcase_87 AC 132 ms
34,892 KB
testcase_88 AC 134 ms
34,888 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;
using Int = long long;
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;
  }
  
};

struct FastIO{
  FastIO(){
    cin.tie(0);
    ios::sync_with_stdio(0);
  }
}fastio_beet;

//INSERT ABOVE HERE

const Int MAX = 314514;
Int pr[MAX];
signed main(){
  for(Int i=2;i<MAX;i++) pr[i]=1;
  for(Int i=2;i<MAX;i++)
    if(pr[i]) for(Int j=i+i;j<MAX;j+=i) pr[j]=0;
  pr[2]=0;

  Int n;
  cin>>n;
  vector<Int> v(n+1),w(2*n+1,0);
  for(Int i=0;i<=n;i++) v[i]=pr[i];
  for(Int i=0;i<=n;i++) w[i*2]=pr[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(!pr[i]) continue;
    ans+=y[i];
    ans-=z[i]*3;
  }
  
  cout<<ans/6<<endl;
  return 0;
}
0