結果
問題 | No.2356 Back Door Tour in Four Seasons |
ユーザー |
![]() |
提出日時 | 2023-07-12 15:30:30 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 199 ms / 2,000 ms |
コード長 | 2,710 bytes |
コンパイル時間 | 1,553 ms |
コンパイル使用メモリ | 142,472 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-09-14 05:12:27 |
合計ジャッジ時間 | 7,508 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 28 |
ソースコード
#include<iostream> #include<vector> #include<string> #include<algorithm> #include<set> #include<queue> #include<cassert> #include<cmath> #include<iomanip> #include<map> #include<functional> //#include<bits/stdc++.h> using namespace std; using ll = long long; const ll mod = 998'244'353; //const ll mod = 1'000'000'007; //const ll mod = 67'280'421'310'721; struct mint{ long long x; mint(long long x=0):x((x%mod+mod)%mod){} mint operator-() const{ return mint(-x); } mint& operator+=(const mint& a){ if((x+=a.x)>=mod)x-=mod; return *this; } mint& operator-=(const mint& a){ if((x+=mod-a.x)>=mod)x-=mod; return *this; } mint& operator*=(const mint& a){ (x *= a.x) %= mod; return *this; } mint operator+(const mint& a) const{ mint res(*this); return res+=a; } mint operator-(const mint& a) const{ mint res(*this); return res-=a; } mint operator*(const mint& a) const{ mint res(*this); return res*=a; } mint pow(long long n) const { assert(0 <= n); mint a = *this, r = 1; while (n) { if (n & 1) r *= a; a *= a; n >>= 1; } return r; } mint inv() const{ return pow(mod-2); } mint& operator/=(const mint& a){ return (*this)*=a.inv(); } mint operator/(const mint& a) const { mint res(*this); return res/=a; } friend ostream& operator<<(ostream& os, const mint& m){ os << m.x; return os; } bool operator==(const mint& a) const { return x == a.x; } bool operator<(const mint& a) const{ return x < a.x; } }; int main(){ int n; cin>>n; vector<char> s(n); vector<int> a(n); for(int i = 0;i<n;i++) cin>>s[i]>>a[i]; vector<mint> tmp(4,0); vector<ll> cnt(4,0); for(int i = 0;i<n;i++){ if(s[i]=='U') {cnt[0] += a[i];} else if(s[i]=='F') cnt[1] += a[i]; else if(s[i]=='W') cnt[2] += a[i]; else cnt[3] += a[i]; } for(int i = 0;i<n;i++){ int ni = 0; if(s[i]=='U'){ ni = 0; }else if(s[i]=='F'){ ni = 1; }else if(s[i]=='W') { ni = 2; }else{ ni = 3; } if(ni==3){ tmp[ni] += mint(n-1).pow(cnt[ni]); continue; } mint other = mint(n-1).pow(cnt[ni]-a[i]); mint now = mint(n-1).pow(a[i]) - mint(n-2).pow(a[i]); tmp[ni] += now * other; } mint ans = 1; for(int i = 0;i<4;i++) ans *= tmp[i]; cout<<ans<<endl; }