結果

問題 No.1079 まお
ユーザー tko919
提出日時 2020-05-12 16:28:19
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,024 ms / 2,000 ms
コード長 2,686 bytes
コンパイル時間 2,565 ms
コンパイル使用メモリ 202,124 KB
実行使用メモリ 26,516 KB
最終ジャッジ日時 2024-10-01 22:45:11
合計ジャッジ時間 19,625 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

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,ll> 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+1,rb);
   unordered_map<int,vector<T>> num; unordered_map<int,vector<ll>> mis,rui;
   unordered_map<int,vector<P>> 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;
      if(cnt==1 and a[i]+a[mid]==k)res+=i-mid+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,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({1,t.second.first});
         else is_one[val].push_back({0,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].first+=is_one[val][i].first;
         is_one[val][i+1].second+=is_one[val][i].second;
      }
   }
   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])-1LL*(i-1)*(rui[k-a[i]].size()-cen);
      }
      cen=lower_bound(ALL(mis[k-a[i]]),mi)-mis[k-a[i]].begin()-1;
      if(cen>=1)res+=is_one[k-a[i]][cen].second-1LL*(i-1)*is_one[k-a[i]][cen].first;
      /*
      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(){
   //ifstream ifs("test/task_0.txt");
   cin>>n>>k; rep(i,0,n)cin>>a[i];
   //ifs>>n>>k; rep(i,0,n)ifs>>a[i];
   rec(0,n); cout<<res<<endl;
   return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0