結果
問題 | No.121 傾向と対策:門松列(その2) |
ユーザー | goodbaton |
提出日時 | 2015-08-08 01:35:38 |
言語 | C++11 (gcc 11.4.0) |
結果 |
AC
|
実行時間 | 3,329 ms / 5,000 ms |
コード長 | 3,514 bytes |
コンパイル時間 | 1,574 ms |
コンパイル使用メモリ | 83,352 KB |
実行使用メモリ | 105,952 KB |
最終ジャッジ日時 | 2023-09-11 04:07:51 |
合計ジャッジ時間 | 8,359 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 57 ms
18,284 KB |
testcase_01 | AC | 99 ms
19,228 KB |
testcase_02 | AC | 9 ms
16,188 KB |
testcase_03 | AC | 503 ms
20,988 KB |
testcase_04 | AC | 3,329 ms
105,952 KB |
testcase_05 | AC | 507 ms
21,016 KB |
testcase_06 | AC | 214 ms
19,900 KB |
testcase_07 | AC | 317 ms
20,016 KB |
testcase_08 | AC | 563 ms
21,056 KB |
ソースコード
#include <cstdio> #include <cstdlib> #include <iostream> #include <string> #include <cmath> #include <algorithm> #include <vector> #include <queue> #include <stack> #include <map> #include <set> #include <cstring> typedef long long ll; using namespace std; #define mod 1000000007 #define INF 1000000000 #define LLINF 2000000000000000000LL #define PI 3.1415926536 #define SIZE 1000002 struct SEG{ int segn2,seg[2<<20]; void init(int n){ segn2=1; while(segn2<n) segn2*=2; //memset(seg,0,sizeof(seg[0])*segn2*2); /* for(int i=0;i<segn2*2-1;i++) seg[i]=0; */ } int q(int a,int b,int l=0,int r=-1,int k=0){ if(r==-1) r+=segn2; if(a<=l && r<=b) return seg[k]; if(r<a || b<l) return 0; return q(a,b,l,(l+r)/2,k*2+1)+q(a,b,(l+r)/2+1,r,k*2+2); } void s(int k,int x){ k+=segn2-1; seg[k]=x; while(k>0){ k=(k-1)/2; seg[k]=seg[k*2+1]+seg[k*2+2]; } return; } void add(int k,int x){ k+=segn2-1; seg[k]+=x; while(k>0){ k=(k-1)/2; seg[k]=seg[k*2+1]+seg[k*2+2]; } return; } }; struct SEGll{ int segn2; ll seg[2<<20]; void init(int n){ segn2=1; while(segn2<n) segn2*=2; //memset(seg,0,sizeof(seg[0])*segn2*2); /* for(int i=0;i<segn2*2-1;i++) seg[i]=0; */ } ll q(int a,int b,int l=0,int r=-1,int k=0){ if(r==-1) r+=segn2; if(a<=l && r<=b) return seg[k]; if(r<a || b<l) return 0; return q(a,b,l,(l+r)/2,k*2+1)+q(a,b,(l+r)/2+1,r,k*2+2); } void s(int k,ll x){ k+=segn2-1; seg[k]=x; while(k>0){ k=(k-1)/2; seg[k]=seg[k*2+1]+seg[k*2+2]; } return; } void add(int k,ll x){ k+=segn2-1; seg[k]+=x; while(k>0){ k=(k-1)/2; seg[k]=seg[k*2+1]+seg[k*2+2]; } return; } }; SEG seg1,seg2; SEGll seg3; int n,a[SIZE],s_a[SIZE],dic_size,d,xa,xb,ya,yb; ll yc,xc; int s1[SIZE],s2[SIZE]; ll s3[SIZE]; ll ans=0,sum3=0; map<int,int> dic; int main(){ scanf("%d",&n); for(int i=0;i<n;i++){ scanf("%d",&a[i]); s_a[i]= a[i]; } sort(s_a,s_a+n); int* end = unique(s_a,s_a+n); dic_size = (int)(end-s_a); //cerr << dic_size << endl; for(int i=0;i<dic_size;i++){ dic[s_a[i]] = i+1; } seg1.init(dic_size+2); seg2.init(dic_size+2); seg3.init(dic_size+2); for(int i=0;i<n;i++){ a[i] = dic[a[i]]; s1[a[i]]++; } for(int i=1;i<=dic_size;i++){ seg1.s(i,s1[i]); } for(int i=0;i<n;i++){ d = a[i]; xa = seg1.q(0,d-1); xb = seg2.q(0,d-1); xc = seg3.q(0,d-1); ya = (n-i)-xa-s1[d]; yb = i-xb-s2[d]; yc = sum3-xc-s3[d]; ans += (ll)xa*xb-xc; ans += (ll)ya*yb-yc; s1[d]--; s2[d]++; sum3+=(ll)s1[d]*s2[d]-s3[d]; s3[d]=(ll)s1[d]*s2[d]; seg1.s(d,s1[d]); seg2.s(d,s2[d]); seg3.s(d,s3[d]); } printf("%lld\n",ans); return 0; }