結果

問題 No.121 傾向と対策：門松列（その２）
コンテスト
ユーザー goodbaton
提出日時 2015-08-08 01:35:38
言語 C++11（廃止可能性あり）
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 2,907 ms / 5,000 ms
コード長 3,514 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 892 ms
コンパイル使用メモリ 85,236 KB
実行使用メモリ 105,216 KB
最終ジャッジ日時 2024-06-28 18:38:36
合計ジャッジ時間 7,285 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 9
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:137:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  137 |     scanf("%d",&n);
      |     ~~~~~^~~~~~~~~
main.cpp:140:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  140 |         scanf("%d",&a[i]);
      |         ~~~~~^~~~~~~~~~~~

ソースコード

diff #
raw source code 

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