結果

問題 No.121 傾向と対策：門松列（その２）
コンテスト
ユーザー chocorusk
提出日時 2019-12-24 04:50:49
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 907 ms / 5,000 ms
コード長 2,194 bytes
コンパイル時間 1,147 ms
コンパイル使用メモリ 120,956 KB
実行使用メモリ 57,856 KB
最終ジャッジ日時 2024-09-19 13:33:44
合計ジャッジ時間 5,489 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <time.h>
#include <stack>
#include <array>
#define popcount __builtin_popcount
using namespace std;
typedef long long int ll;
typedef pair<int, int> P;
template<typename T>
struct BIT{ //1-indexed
    vector<T> bit;
    int size;
    BIT(int n):size(n), bit(n+1, 0){}
	void init(){
		fill(bit.begin(), bit.end(), 0);
	}
    T sum(int i){
        T s=0;
        while(i>0){
            s+=bit[i];
            i-=(i&(-i));
        }
        return s;
    }
    void add(int i, T x){
        while(i<=size){
            bit[i]+=x;
            i+=(i&(-i));
        }
    }
};
int main()
{
	int n;
	cin>>n;
	int a[1000010];
	vector<int> v(n);
	for(int i=0; i<n; i++){
		scanf("%d", &a[i]);
		v[i]=a[i];
	}
	sort(v.begin(), v.end());
	v.erase(unique(v.begin(), v.end()), v.end());
	for(int i=0; i<n; i++){
		a[i]=lower_bound(v.begin(), v.end(), a[i])-v.begin();
	}
	int m=v.size();
	BIT<ll> bit(m), bit2(m);
	ll c1[1000010], c2[1000010], c[1000010]={}, c0[1000010]={};
	for(int i=0; i<n; i++) c[a[i]]++;
	ll ans=0;
	for(int i=0; i<n; i++){
		c1[i]=bit.sum(a[i]);
		ans-=bit2.sum(a[i]);
		bit2.add(a[i]+1, -c0[a[i]]*(c[a[i]]-c0[a[i]]));
		c0[a[i]]++;
		bit2.add(a[i]+1, c0[a[i]]*(c[a[i]]-c0[a[i]]));
		bit.add(a[i]+1, 1);
	}
	bit.init();
	for(int i=n-1; i>=0; i--){
		c2[i]=bit.sum(a[i]);
		bit.add(a[i]+1, 1);
	}
	for(int i=0; i<n; i++){
		ans+=c1[i]*c2[i];
	}
	bit.init();
	bit2.init();
	fill(c0, c0+m, 0);
	for(int i=0; i<n; i++){
		c1[i]=i-bit.sum(a[i]+1);
		ans-=bit2.sum(m)-bit2.sum(a[i]+1);
		bit2.add(a[i]+1, -c0[a[i]]*(c[a[i]]-c0[a[i]]));
		c0[a[i]]++;
		bit2.add(a[i]+1, c0[a[i]]*(c[a[i]]-c0[a[i]]));
		bit.add(a[i]+1, 1);
	}
	bit.init();
	for(int i=n-1; i>=0; i--){
		c2[i]=(n-1-i)-bit.sum(a[i]+1);
		bit.add(a[i]+1, 1);
	}
	for(int i=0; i<n; i++){
		ans+=c1[i]*c2[i];
	}
	cout<<ans<<endl;
	return 0;
}
0