1. 程式人生 > >P1311 選擇客棧

P1311 選擇客棧

cto string urn http bsp 每一個 lld spa void

P1311 選擇客棧

RMQ+二分(當然這並不是正解)

對於每一個客棧,二分滿足條件的範圍,在RMQ上查詢。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<cctype>
using namespace std;
template <typename T> inline T min(T &a,T &b) {return a<b ? a:b;}
template <typename T> inline void
read(T &x){ char c=getchar(); x=0; while(!isdigit(c)) c=getchar(); while(isdigit(c)) x=(x<<3)+(x<<1)+(c^48),c=getchar(); } int n,k,p,mnd[18][200002],log[200002]; long long ans; vector <int> g[52]; inline int query(int l,int r){ int k=log[r-l+1]; return min(mnd[k][l],mnd[k][r-(1
<<k)+1]); } int main(){ read(n); read(k); read(p); int q; log[0]=-1; for(int i=1;i<=n;++i){ read(q); read(mnd[0][i]); g[q].push_back(i); log[i]=log[i>>1]+1; } for(int i=1;(1<<i)<=n;++i) for(int j=1;j+(1<<(i-1))<=n;++j) mnd[i][j]
=min(mnd[i-1][j],mnd[i-1][j+(1<<(i-1))]); for(int i=0;i<k;++i) for(int j=0;j<g[i].size()-1;++j){ if(query(g[i][j],g[i][g[i].size()-1])>p) break; //已經無法滿足條件就跳出 int l=j+1,r=g[i].size()-1,mid; while(l<r){ //二分確定範圍 mid=l+((r-l)>>1); if(query(g[i][j],g[i][mid])>p) l=mid+1; else r=mid; }ans+=g[i].size()-l; } printf("%lld",ans); return 0; }

P1311 選擇客棧