結果
問題 | No.1193 Penguin Sequence |
ユーザー | 👑 potato167 |
提出日時 | 2022-05-10 21:15:04 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 859 ms / 2,000 ms |
コード長 | 4,642 bytes |
コンパイル時間 | 2,688 ms |
コンパイル使用メモリ | 228,208 KB |
実行使用メモリ | 164,012 KB |
最終ジャッジ日時 | 2023-09-25 04:39:15 |
合計ジャッジ時間 | 27,280 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 476 ms
163,848 KB |
testcase_01 | AC | 825 ms
163,916 KB |
testcase_02 | AC | 859 ms
163,516 KB |
testcase_03 | AC | 841 ms
163,516 KB |
testcase_04 | AC | 839 ms
163,836 KB |
testcase_05 | AC | 854 ms
163,492 KB |
testcase_06 | AC | 839 ms
164,012 KB |
testcase_07 | AC | 837 ms
163,516 KB |
testcase_08 | AC | 842 ms
163,900 KB |
testcase_09 | AC | 821 ms
163,808 KB |
testcase_10 | AC | 846 ms
163,492 KB |
testcase_11 | AC | 418 ms
93,812 KB |
testcase_12 | AC | 420 ms
95,140 KB |
testcase_13 | AC | 682 ms
141,604 KB |
testcase_14 | AC | 627 ms
132,636 KB |
testcase_15 | AC | 764 ms
154,024 KB |
testcase_16 | AC | 83 ms
9,372 KB |
testcase_17 | AC | 2 ms
4,376 KB |
testcase_18 | AC | 41 ms
16,284 KB |
testcase_19 | AC | 827 ms
161,216 KB |
testcase_20 | AC | 615 ms
128,140 KB |
testcase_21 | AC | 467 ms
103,656 KB |
testcase_22 | AC | 52 ms
19,672 KB |
testcase_23 | AC | 393 ms
87,296 KB |
testcase_24 | AC | 338 ms
76,384 KB |
testcase_25 | AC | 133 ms
36,856 KB |
testcase_26 | AC | 24 ms
11,500 KB |
testcase_27 | AC | 676 ms
138,396 KB |
testcase_28 | AC | 442 ms
97,440 KB |
testcase_29 | AC | 699 ms
141,472 KB |
testcase_30 | AC | 193 ms
51,172 KB |
testcase_31 | AC | 149 ms
42,752 KB |
testcase_32 | AC | 513 ms
112,376 KB |
testcase_33 | AC | 334 ms
79,504 KB |
testcase_34 | AC | 257 ms
65,100 KB |
testcase_35 | AC | 329 ms
80,632 KB |
testcase_36 | AC | 250 ms
63,248 KB |
testcase_37 | AC | 646 ms
135,400 KB |
testcase_38 | AC | 2 ms
4,376 KB |
testcase_39 | AC | 2 ms
4,380 KB |
testcase_40 | AC | 2 ms
4,380 KB |
ソースコード
#include <bits/stdc++.h> #pragma GCC optimize("Ofast") #define _GLIBCXX_DEBUG using namespace std; using std::cout; using std::cin; using std::endl; using ll=long long; using ld=long double; ll ILL=2167167167167167167; const int INF=2100000000; const ll mod=998244353; #define rep(i,a) for (ll i=0;i<a;i++) #define all(p) p.begin(),p.end() template<class T> using _pq = priority_queue<T, vector<T>, greater<T>>; template<class T> ll LB(vector<T> &v,T a){return lower_bound(v.begin(),v.end(),a)-v.begin();} template<class T> ll UB(vector<T> &v,T a){return upper_bound(v.begin(),v.end(),a)-v.begin();} template<class T> bool chmin(T &a,const T &b){if(a>b){a=b;return 1;}else return 0;} template<class T> bool chmax(T &a,const T &b){if(a<b){a=b;return 1;}else return 0;} template<class T> void So(vector<T> &v) {sort(v.begin(),v.end());} template<class T> void Sore(vector<T> &v) {sort(v.begin(),v.end(),[](T x,T y){return x>y;});} void yneos(bool a){if(a) cout<<"Yes\n"; else cout<<"No\n";} template<class T> void vec_out(vector<T> &p){for(int i=0;i<(int)(p.size());i++){if(i) cout<<" ";cout<<p[i];}cout<<"\n";} namespace po167{ template <class T> struct seg_sum{ int n; int seg_size; std::vector<T> seg; T e; //要素数、単位元 typeは+演算が定義されているもの seg_sum(int k,T e_in){ e=e_in; n=k; seg_size=1; while(seg_size<n) seg_size*=2; seg.resize(seg_size*2-1); for(int i=0;i<seg_size*2-1;i++) seg[i]=e; } T sum_sub(int a,int b,int l,int r,int k){ if(r<=a||b<=l) return e; if(a<=l&&r<=b) return seg[k]; T X=sum_sub(a,b,l,(l+r)/2,k*2+1); T Y=sum_sub(a,b,(l+r)/2,r,k*2+2); return X+Y; } T sum(int a,int b){ //assert(0<=a&&b<n); return sum_sub(a,b,0,seg_size,0); } T get(int index){ assert(0<=index&&index<n); return seg[seg_size-1+index]; } void insert(int index,T a){ assert(0<=index&&index<n); int now_index=index+seg_size-1; seg[now_index]=a; while(now_index!=0){ now_index-=1; now_index/=2; seg[now_index]=seg[now_index*2+1]+seg[now_index*2+2]; } } }; template <class T> //p,qのなかみは、重複ありで内容が同じならおけ long long inversion(std::vector<T> &p,std::vector<T> &q){ int n=p.size(); assert(p.size()==q.size()); po167::seg_sum<int> S(n,0); std::vector<int> p_index(n,-1); std::map<T,std::queue<int>> m; for(int i=0;i<n;i++) m[p[i]].push(i); for(int i=0;i<n;i++){ if(m[q[i]].empty()) return -1; int a=m[q[i]].front(); m[q[i]].pop(); p_index[i]=a; } long long ans=0; for(int i=0;i<n;i++){ ans+=S.sum(p_index[i],n); S.insert(p_index[i],1); } return ans; } template <class T> //qのなかみは、0~nの順列の並び替えである必要がある long long inversion(std::vector<T> &q){ int n=q.size(); po167::seg_sum<int> S(n,0); std::vector<int> p_index(n); for(int i=0;i<n;i++){ p_index[i]=i; } long long ans=0; for(int i=0;i<n;i++){ assert(0<=q[i]&&q[i]<n); assert(S.get(q[i])==0); ans+=S.sum(q[i],n); S.insert(q[i],1); } return ans; } } using po167::inversion; using po167::seg_sum; namespace po167{ struct combination{ int upper; long long MOD; std::vector<long long> fact; std::vector<long long> rev; std::vector<long long> fact_rev; combination(int max,long long mod):upper(max),MOD(mod),fact(max+1),rev(max+1),fact_rev(max+1){ for(long long i=0;i<=max;i++){ if(i<2){ fact[i]=1; fact_rev[i]=1; rev[i]=1; continue; } fact[i]=(fact[i-1]*i)%mod; rev[i]=mod-((mod/i)*rev[mod%i])%mod; fact_rev[i]=(fact_rev[i-1]*rev[i])%mod; } } long long Comb(int x,int y){ assert(upper>=x); if (x<y||y<0||x<0) return 0; return (((fact_rev[y]*fact_rev[x-y])%MOD)*fact[x])%MOD; } }; } using po167::combination; ll jyo(ll x,ll y,ll z){ ll H=y; //ここから ll a=1,b=(x%z+z)%z,c=1; while(H>0){ a*=2; if(H%a!=0){ H-=a/2; c*=b; c%=z; } b*=b; b%=z; } //ここまで return c; } void solve(); // oddloop int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int t=1; //cin>>t; rep(i,t) solve(); } void solve(){ ll N; cin>>N; vector<int> A(N); ll X=0; map<ll,int> m; rep(i,N){ cin>>A[i]; X+=i-m[A[i]]; m[A[i]]++; } if(N==1){ cout<<"0\n"; return ; } auto B=A; So(B); ll I=inversion(A,B); ll ans=N+1; ans=(ans*jyo(3,mod-2,mod))%mod; ans=(ans*I)%mod; combination table(N,mod); ll S=0; rep(i,N) S=(S+i+1)%mod; S=(S*S)%mod; rep(i,N) S=(S-(i+1)*(i+1))%mod; S=(S*jyo(2ll*N*N,mod-2,mod))%mod; S=(S*(X%mod))%mod; ans=(ans+S+mod)%mod; rep(i,N){ ans=(ans*table.Comb(N,i+1))%mod; } cout<<ans<<"\n"; }