結果
問題 | No.805 UMG |
ユーザー |
![]() |
提出日時 | 2019-03-17 15:29:05 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,825 bytes |
コンパイル時間 | 1,279 ms |
コンパイル使用メモリ | 106,908 KB |
実行使用メモリ | 6,824 KB |
最終ジャッジ日時 | 2024-11-18 13:09:09 |
合計ジャッジ時間 | 2,213 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 5 WA * 20 |
ソースコード
#include<iostream> #include<string> #include<algorithm> #include<vector> #include<iomanip> #include<math.h> #include<complex> #include<queue> #include<deque> #include<stack> #include<map> #include<set> #include<bitset> #include<functional> #include<assert.h> #include<numeric> using namespace std; #define REP(i,m,n) for(int i=(int)(m) ; i < (int) (n) ; ++i ) #define rep(i,n) REP(i,0,n) typedef long long ll; typedef pair<int,int> pint; typedef pair<ll,int> pli; const int inf=1e9+7; const ll longinf=1LL<<60 ; const ll mod=1e9+7 ; typedef vector<complex<double>> poly; const double pi=acos(-1); poly dft(poly f,int rev){ int n=f.size(); int j=0; REP(i,1,n-1){ for(int k=n/2;k>(j^=k);k>>=1); if(i<j)swap(f[i],f[j]); } complex<double> zeta,ret,t,s; for(int i=1;i<n;i<<=1){ zeta=polar(1.0,pi/i*rev); for(int j=0;j<n;j+=2*i){ ret=1.0; rep(k,i){ s=f[j+k]; t=f[j+k+i]; t=complex<double>(t.real()*ret.real()-t.imag()*ret.imag(),t.imag()*ret.real()+t.real()*ret.imag()); f[j+k]=s+t; f[j+k+i]=s-t; ret*=zeta; } } } if(rev==-1)rep(i,n)f[i]/=n; return f; } poly fft(poly g,poly h){ poly f; int m=(int)g.size()+h.size()+1; int sz=1; while(sz<m)sz*=2; f.resize(sz,0); g.resize(sz,0); h.resize(sz,0); g=dft(g,1); h=dft(h,1); rep(i,sz)f[i]=g[i]*h[i]; f=dft(f,-1); return f; } int main(){ int n; cin>>n; string s; cin>>s; poly u(n),g(n); rep(i,n){ if(s[i]=='U')u[i]=1; if(s[i]=='G')g[i]=1; } auto ret=fft(u,g); ll ans=0; rep(i,n){ if(s[i]=='M'){ int res=(ret[2*i].real()+.5); ans+=res; } } cout<<ans<<endl; return 0; }