結果

問題 No.121 傾向と対策：門松列（その２）
コンテスト
ユーザー goodbaton
提出日時 2015-08-08 01:17:18
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 1,623 ms / 5,000 ms
コード長 2,442 bytes
コンパイル時間 725 ms
コンパイル使用メモリ 83,072 KB
実行使用メモリ 89,600 KB
最終ジャッジ日時 2024-06-28 18:37:56
合計ジャッジ時間 5,158 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 9
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:95:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   95 |     scanf("%d",&n);
      |     ~~~~~^~~~~~~~~
main.cpp:98:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   98 |         scanf("%d",&a[i]);
      |         ~~~~~^~~~~~~~~~~~

ソースコード

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

/* BIT(1-index) */
struct BIT{
    int BIT_n;

    int data[SIZE+1];
    
    BIT(int n){
        memset(data,0,sizeof(data));
        BIT_n = n;
    }
    
    void add(int k,int x){
        while(k<=BIT_n){
            data[k]+=x;
            k+= k&(-k);
        }
    }

    int query(int k){
        int rec=0;
        
        while(k>0){
            rec+=data[k];
            k-= k&(-k);
        }
        
        return rec;
    }
};

struct BITll{
    ll BIT_n;
    
    ll data[SIZE+1];
    
    BITll(int n){
        memset(data,0,sizeof(data));
        BIT_n = n;
    }
    
    void add(int k,ll x){
        while(k<=BIT_n){
            data[k]+=x;
            k+= k&(-k);
        }
    }
    
    ll query(int k){
        ll rec=0;
        
        while(k>0){
            rec+=data[k];
            k-= k&(-k);
        }
        
        return rec;
    }
};

int n,a[SIZE],s_a[SIZE],dic_size,d;
ll xa,xb,xc=0,ya,yb,yc=0;
int s1[SIZE],s2[SIZE];
ll s3[SIZE];
ll sum3=0;
ll ans=0;
map<int,int> dic;
BIT bit1(SIZE),bit2(SIZE);
BITll bit3(SIZE);

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);
    
    for(int i=0;i<dic_size;i++){
        dic[s_a[i]] = i+1;
    }
    
    for(int i=0;i<n;i++){
        a[i] = dic[a[i]];
        s1[a[i]]++;
    }
    
    for(int i=1;i<=dic_size;i++){
        bit1.add(i,s1[i]);
    }
    
    for(int i=0;i<n;i++){
        d = a[i];
        
        xa = bit1.query(d-1);
        xb = bit2.query(d-1);
        xc = bit3.query(d-1);
        ya = (n-i)-xa-s1[d];
        yb = i-xb-s2[d];
        yc = sum3-xc-s3[d];
        
        ans += xa*xb-xc;
        ans += ya*yb-yc;
        
        s1[d]--;
        s2[d]++;
        bit1.add(d,-1);
        bit2.add(d,1);
        bit3.add(d,(ll)s1[d]*s2[d]-s3[d]);
        sum3+=(ll)s1[d]*s2[d]-s3[d];
        s3[d]=(ll)s1[d]*s2[d];
    }
    
    printf("%lld\n",ans);
    
    return 0;
}
0