結果

問題 No.121 傾向と対策：門松列（その２）
ユーザー goodbatongoodbaton
提出日時 2015-08-07 21:44:21
言語 C++11
(gcc 11.4.0)
結果
TLE  
実行時間 -
コード長 2,426 bytes
コンパイル時間 753 ms
コンパイル使用メモリ 84,812 KB
実行使用メモリ 94,400 KB
最終ジャッジ日時 2023-09-25 05:58:49
合計ジャッジ時間 11,498 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 97 ms
17,924 KB
testcase_01 AC 165 ms
18,952 KB
testcase_02 AC 12 ms
16,028 KB
testcase_03 AC 971 ms
20,940 KB
testcase_04 TLE -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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;
        
        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;
    }
};

SEG seg1,seg2,seg3;

int n,a[SIZE],s_a[SIZE],dic_size,d,xa,xb,xc,ya,yb,yc;
int s1[SIZE],s2[SIZE],s3[SIZE];
ll ans=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++){
        d = dic[a[i]];
        seg1.add(d,1);
        s1[d]++;
    }
    
    for(int i=0;i<n;i++){
        d = dic[a[i]];
        
        xa = seg1.q(0,d-1);
        xb = seg2.q(0,d-1);
        xc = seg3.q(0,d-1);
        ya = seg1.q(d+1,dic_size+1);
        yb = seg2.q(d+1,dic_size+1);
        yc = seg3.q(d+1,dic_size+1);
        
        ans += (ll)xa*xb-xc;
        ans += (ll)ya*yb-yc;
        
        s1[d]--;
        s2[d]++;
        s3[d]=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;
}
0