結果

問題 No.1079 まお
ユーザー Chanyuh
提出日時 2020-07-22 11:28:21
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 2,383 bytes
コンパイル時間 1,662 ms
コンパイル使用メモリ 119,972 KB
実行使用メモリ 54,500 KB
最終ジャッジ日時 2024-06-11 07:37:45
合計ジャッジ時間 26,606 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18 WA * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<string>
#include<cstdio>
#include<vector>
#include<cmath>
#include<algorithm>
#include<functional>
#include<iomanip>
#include<queue>
#include<ciso646>
#include<random>
#include<map>
#include<set>
#include<complex>
#include<bitset>
#include<stack>
#include<unordered_map>
#include<utility>
#include<tuple>
using namespace std;
typedef long long ll;
typedef unsigned int ui;
const ll mod = 1000000007;
const ll INF = (ll)1000000007 * 1000000007;
typedef pair<int, int> P;
#define stop char nyaa;cin>>nyaa;
#define rep(i,n) for(int i=0;i<n;i++)
#define per(i,n) for(int i=n-1;i>=0;i--)
#define Rep(i,sta,n) for(int i=sta;i<n;i++)
#define Per(i,sta,n) for(int i=n-1;i>=sta;i--)
#define rep1(i,n) for(int i=1;i<=n;i++)
#define per1(i,n) for(int i=n;i>=1;i--)
#define Rep1(i,sta,n) for(int i=sta;i<=n;i++)
typedef long double ld;
const ld eps = 1e-8;
const ld pi = acos(-1.0);
typedef pair<ll, ll> LP;
typedef pair<P, ll> PP;
int dx[4]={1,-1,0,0};
int dy[4]={0,0,1,-1};

int n,k;
int a[100010];

ll rec(int l,int r){
  if(r-l<=0) return 0;
  if(r-l==1){
    if(2*a[l]==k) return 1;
    else return 0;
  }
  map<int,ll> ma1,ma3;
  map<P,ll> ma2,ma4;
  int mid=(l+r)/2;
  vector<P> v={};
  P p=P(mod,0);
  Per(i,l,mid){
    int mi=p.first,cnt=p.second;
    if(mi>a[i]){
      mi=a[i];
      cnt=1;
    }
    else if(mi==a[i]) cnt+=1;
    //cout << mi << " " << cnt << endl;
    if(cnt==1) v.push_back(P(mi,i));
    p=P(mi,cnt);
  }
  p=P(mod,0);
  Rep(i,mid,r){
    int mi=p.first,cnt=p.second;
    if(mi>a[i]){
      mi=a[i];
      cnt=1;
    }
    else if(mi==a[i]) cnt+=1;
    if(cnt==1) {
      ma1[a[i]]+=i-mid+1;
      ma2[P(a[i],mi)]+=i-mid+1;
      ma3[a[i]]+=1;
      ma4[P(a[i],mi)]+=1;
    }
    p=P(mi,cnt);
  }
  ll res=0;
  for(P s:v){
    int mi=s.first,id=s.second;
    //cout << mi << " " << id << endl;
    //cout << ma1[k-a[id]]+(mid-id)*ma3[k-a[id]] << endl;
    //cout << ma2[P(k-a[id],mi)]+(mid-id)*ma4[P(k-a[id],mi)] << endl;
    res+=ma1[k-a[id]]+(mid-id)*ma3[k-a[id]];
    res-=ma2[P(k-a[id],mi)]+(mid-id)*ma4[P(k-a[id],mi)];
  }
  //cout << l << " " << r << endl;
  //cout << res << endl;
  return res+rec(l,mid)+rec(mid,r);
}

void solve(){
  cin >> n >> k;
  rep(i,n){
    cin >> a[i];
  }
  cout << rec(0,n) << endl;
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout << fixed << setprecision(50);
    solve();
}
0