結果

問題 No.1079 まお
ユーザー Y17Y17
提出日時 2020-06-12 23:31:30
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 340 ms / 2,000 ms
コード長 2,671 bytes
コンパイル時間 1,880 ms
コンパイル使用メモリ 184,456 KB
実行使用メモリ 10,412 KB
最終ジャッジ日時 2023-09-06 11:26:53
合計ジャッジ時間 7,840 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 1 ms
4,380 KB
testcase_04 AC 2 ms
4,376 KB
testcase_05 AC 2 ms
4,376 KB
testcase_06 AC 2 ms
4,376 KB
testcase_07 AC 7 ms
4,380 KB
testcase_08 AC 7 ms
4,380 KB
testcase_09 AC 7 ms
4,376 KB
testcase_10 AC 7 ms
4,376 KB
testcase_11 AC 7 ms
4,380 KB
testcase_12 AC 135 ms
6,688 KB
testcase_13 AC 110 ms
7,120 KB
testcase_14 AC 217 ms
8,164 KB
testcase_15 AC 189 ms
8,780 KB
testcase_16 AC 179 ms
9,376 KB
testcase_17 AC 263 ms
10,160 KB
testcase_18 AC 186 ms
10,264 KB
testcase_19 AC 328 ms
10,068 KB
testcase_20 AC 282 ms
10,072 KB
testcase_21 AC 248 ms
10,216 KB
testcase_22 AC 268 ms
10,412 KB
testcase_23 AC 264 ms
10,256 KB
testcase_24 AC 295 ms
10,248 KB
testcase_25 AC 331 ms
10,216 KB
testcase_26 AC 340 ms
10,252 KB
testcase_27 AC 163 ms
10,164 KB
testcase_28 AC 181 ms
10,256 KB
testcase_29 AC 167 ms
10,412 KB
testcase_30 AC 167 ms
10,132 KB
testcase_31 AC 177 ms
10,264 KB
権限があれば一括ダウンロードができます

ソースコード

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