結果

問題 No.1079 まお
ユーザー Y17
提出日時 2020-06-12 23:31:30
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 349 ms / 2,000 ms
コード長 2,671 bytes
コンパイル時間 2,361 ms
コンパイル使用メモリ 185,692 KB
実行使用メモリ 10,320 KB
最終ジャッジ日時 2024-06-24 06:03:41
合計ジャッジ時間 7,666 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#define int long long
#define mp make_pair
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; }

int a[100010];
int k;

template<class T> struct BinaryIndexTree{
    int n;
    vector<T> bit; 
    vector<T> org;

    BinaryIndexTree(int si): bit(si+1, 0), org(si+1, 0) { n = si; }; 

    void add(int idx, T v){
        org[++idx] += v;
        for(int i = idx;i <= n;i+=i&-i) bit[i] += v;
    }

    void update(int idx, T v){
        T vv = v - org[++idx];
        org[idx] = v;
        for(int i = idx;i <= n;i+=i&-i) bit[i] += vv;
    }

    T sum(int a){
        T ret = 0;
        for(int i = a;i > 0;i-=i&-i) ret += bit[i];
        return ret;
    }

    T sum(int l, int r) { return sum(r)-sum(l); }; 
};


int dfs(int l, int r){
	if(r-l == 1) return a[l]*2 == k ? 1 : 0; 
	int mid = (r+l)/2;
	int ans = dfs(l, mid) + dfs(mid, r);
	vector<pair<int, pair<pair<int, int>,int>>> b(mid-l), c(r-mid);

	for(int i = l;i < mid;i++) b[i-l].first = a[i];
	for(int i = mid;i < r;i++) c[i-mid].first = a[i];

	int ln = b.size();
	int rn = c.size();
	int mi = INT_MAX;
	int micnt = 0;
	for(int i = ln-1;i >= 0;i--){
		if(mi > b[i].first){
			mi = b[i].first;
			micnt = 1;
		}else if(mi == b[i].first){
			micnt++;
		}

		b[i].second = {{mi, micnt}, ln-i};
	}

	sort(b.begin(), b.end());

	BinaryIndexTree<int> bit1_c(ln), bit1_s(ln);
	BinaryIndexTree<int> bit2_c(ln), bit2_s(ln);

	for(int i = 0;i < ln;i++){
		if(b[i].second.first.second == 1){
			bit2_c.update(i, 1);
			bit2_s.update(i, b[i].second.second);
		}
		bit1_c.update(i, 1);
		bit1_s.update(i, b[i].second.second);
	}

	mi = INT_MAX;
	micnt = 0;
	for(int i = 0;i < rn;i++){
		if(mi > c[i].first){
			mi = c[i].first;
			micnt = 1;
		}else if(mi == c[i].first){
			micnt++;
		}
		int ll, m1, m2, rr; 
		pair<int, pair<pair<int, int>, int>> x = mp(k-c[i].first, mp(mp(-1, -1), -1));
		ll = lower_bound(b.begin(), b.end(), x) - b.begin();

		x = mp(k-c[i].first, mp(mp(mi, -1), -1));
		m1 = lower_bound(b.begin(), b.end(), x) - b.begin();

		x = mp(k-c[i].first, mp(mp(mi+1, -1), -1));
		m2 = lower_bound(b.begin(), b.end(), x) - b.begin();

		x = mp(k-c[i].first+1, mp(mp(-1, -1), -1));
		rr = lower_bound(b.begin(), b.end(), x) - b.begin();
		int atmp = bit2_c.sum(ll, m1)*(i+1) + bit2_s.sum(ll, m1);
		if(micnt == 1){
			atmp += bit1_c.sum(m2, rr)*(i+1) + bit1_s.sum(m2, rr);
		}
		ans += atmp;
	}
	return ans;
}

signed main(){
	int n;
	cin >> n >> k;

	for(int i = 0;i < n;i++){
		cin >> a[i];
	}

	cout << dfs(0, n) << endl;
	return 0;
}
0