結果

問題 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;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0