結果

問題 No.404 部分門松列
ユーザー chocoruskchocorusk
提出日時 2020-01-01 14:21:00
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 744 ms / 2,000 ms
コード長 2,643 bytes
コンパイル時間 1,325 ms
コンパイル使用メモリ 123,528 KB
実行使用メモリ 18,260 KB
最終ジャッジ日時 2023-08-14 08:20:52
合計ジャッジ時間 18,302 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
16,024 KB
testcase_01 AC 5 ms
15,440 KB
testcase_02 AC 5 ms
15,884 KB
testcase_03 AC 5 ms
15,584 KB
testcase_04 AC 5 ms
15,184 KB
testcase_05 AC 5 ms
15,168 KB
testcase_06 AC 5 ms
14,920 KB
testcase_07 AC 5 ms
15,220 KB
testcase_08 AC 5 ms
16,168 KB
testcase_09 AC 4 ms
15,840 KB
testcase_10 AC 5 ms
15,456 KB
testcase_11 AC 5 ms
14,704 KB
testcase_12 AC 5 ms
15,888 KB
testcase_13 AC 366 ms
16,084 KB
testcase_14 AC 417 ms
16,504 KB
testcase_15 AC 351 ms
15,948 KB
testcase_16 AC 251 ms
17,408 KB
testcase_17 AC 498 ms
17,508 KB
testcase_18 AC 336 ms
15,912 KB
testcase_19 AC 170 ms
15,924 KB
testcase_20 AC 429 ms
17,016 KB
testcase_21 AC 443 ms
17,192 KB
testcase_22 AC 409 ms
15,852 KB
testcase_23 AC 515 ms
17,212 KB
testcase_24 AC 671 ms
17,080 KB
testcase_25 AC 744 ms
18,260 KB
testcase_26 AC 703 ms
17,768 KB
testcase_27 AC 262 ms
15,400 KB
testcase_28 AC 513 ms
16,188 KB
testcase_29 AC 532 ms
17,752 KB
testcase_30 AC 468 ms
16,488 KB
testcase_31 AC 154 ms
14,932 KB
testcase_32 AC 466 ms
17,596 KB
testcase_33 AC 582 ms
16,872 KB
testcase_34 AC 577 ms
16,408 KB
権限があれば一括ダウンロードができます

ソースコード

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, ll> 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[200020];
	vector<int> v(n);
	for(int i=0; i<n; i++){
		cin>>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);
	ll c1[200020];
	for(int i=0; i<n; i++){
		c1[i]=bit.sum(a[i]);
		bit.add(a[i]+1, 1);
	}
	bit.init();
	ll c2[200020];
	for(int i=n-1; i>=0; i--){
		c2[i]=bit.sum(a[i]);
		bit.add(a[i]+1, 1);
	}
	ll c[200020]={};
	for(int i=0; i<n; i++){
		c[i]+=c1[i]*c2[i];
	}
	ll x[200020]={}, x1[200020]={};
	for(int i=0; i<n; i++) x[a[i]]++;
	bit.init();
	for(int i=0; i<n; i++){
		c[i]-=bit.sum(a[i]);
		bit.add(a[i]+1, -x1[a[i]]*(x[a[i]]-x1[a[i]]));
		x1[a[i]]++;
		bit.add(a[i]+1, x1[a[i]]*(x[a[i]]-x1[a[i]]));
	}
	bit.init();
	for(int i=0; i<n; i++){
		c1[i]=i-bit.sum(a[i]+1);
		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++){
		c[i]+=c1[i]*c2[i];
	}
	fill(x1, x1+m+1, 0);
	bit.init();
	for(int i=0; i<n; i++){
		c[i]-=(bit.sum(m)-bit.sum(a[i]+1));
		bit.add(a[i]+1, -x1[a[i]]*(x[a[i]]-x1[a[i]]));
		x1[a[i]]++;
		bit.add(a[i]+1, x1[a[i]]*(x[a[i]]-x1[a[i]]));
	}
	P p[200020];
	for(int i=0; i<n; i++) p[i]=P(a[i], c[i]);
	sort(p, p+n);
	ll t[200020]; t[0]=0;
	for(int i=0; i<n; i++) t[i+1]=t[i]+p[i].second;
	int q; cin>>q;
	for(int i=0; i<q; i++){
		int l, h; cin>>l>>h; h++;
		l=lower_bound(v.begin(), v.end(), l)-v.begin();
		h=lower_bound(v.begin(), v.end(), h)-v.begin();
		l=lower_bound(p, p+n, P(l, 0))-p;
		h=lower_bound(p, p+n, P(h, 0))-p;
		cout<<t[h]-t[l]<<endl;
	}
 	return 0;
}
0