結果

問題 No.1079 まお
コンテスト
ユーザー tko919
提出日時 2020-05-12 15:10:59
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,385 bytes
コンパイル時間 2,430 ms
コンパイル使用メモリ 192,292 KB
実行使用メモリ 25,092 KB
最終ジャッジ日時 2024-10-01 22:43:18
合計ジャッジ時間 17,334 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 2 WA * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

//template
#define rep(i,a,b) for(int i=(int)(a);i<(int)(b);i++)
#define ALL(v) (v).begin(),(v).end()
typedef long long int ll;
const int inf = 0x3fffffff; const ll INF = 0x1fffffffffffffff; const double eps=1e-12;
template<typename T>inline bool chmax(T& a,T b){if(a<b){a=b;return 1;}return 0;}
template<typename T>inline bool chmin(T& a,T b){if(a>b){a=b;return 1;}return 0;}
//end

typedef pair<int,int> P;
typedef pair<int,P> T;
const int m=101010;
int n,k,a[m]; ll res=0;
void rec(int lb,int rb){
   if(rb-lb<=2){
      if(rb-lb==1 and a[lb]*2==k)res++;
      else if(rb-lb==2){
         if(a[lb]+a[lb+1]==k and a[lb]!=a[lb+1])res+=2;
         if(a[lb]*2==k)res++;
         if(a[lb+1]*2==k)res++;
      }
      return;
   }
   int mid=(lb+rb)/2; rec(lb,mid); rec(mid,rb);
   unordered_map<int,vector<T>> num; unordered_map<int,vector<int>> mis,rui,is_one;
   int mi=inf,cnt=1;
   for(int i=mid;i<rb;i++){
      if(mi==a[i])cnt++;
      if(chmin(mi,a[i]))cnt=1;
      num[a[i]].push_back({mi,{i,cnt}});
   }
   for(auto p:num){
      int val=p.first; vector<T> v=p.second;
      sort(ALL(v));
      mis[val].push_back(-inf);
      rui[val].push_back(0); is_one[val].push_back(0);
      for(T t:v){
         mis[val].push_back(t.first);
         rui[val].push_back(t.second.first);
         if(t.second.second==1)is_one[val].push_back(t.second.first);
         else is_one[val].push_back(0);
      }
      mis[val].push_back(inf);
      rep(i,0,rui[val].size()-1){
         rui[val][i+1]+=rui[val][i]; is_one[val][i+1]+=is_one[val][i];
      }
   }
   mi=inf; cnt=1;
   for(int i=mid-1;i>=lb;i--){
      if(mi==a[i])cnt++;
      if(chmin(mi,a[i]))cnt=1;
      if(mis[k-a[i]].empty())continue;
      int cen=upper_bound(ALL(mis[k-a[i]]),mi)-mis[k-a[i]].begin();
      if(cnt==1 and mis[k-a[i]][cen]!=inf){
         res+=(*rui[k-a[i]].rbegin()-rui[k-a[i]][cen-1])-i*(rui[k-a[i]].size()-cen-1);
      }
      cen--; if(mis[k-a[i]][cen]==mi)cen--;
      if(cen>=1)res+=is_one[k-a[i]][cen]-i*(cen+1);
      /*
      for(auto p:num[k-a[i]]){
         int opp=p.first,idx=p.second.first,add=p.second.second;
         if(opp<mi and add==1)res+=idx-i+1;
         else if(opp>mi and cnt==1)res+=idx-i+1;
      }
      */
   }
}

int main(){
   cin>>n>>k; rep(i,0,n)cin>>a[i];
   rec(0,n); cout<<res<<endl;
   return 0;
}
0