[codeforce 1405]E. Fixed Point Removal树状数组

参考文章Link
1.对每个点可以处理出在这个点之前至多可以删除的个数,使用二分搜索来处理,对一个可删除个数a
if query(i - 1) - query(a - 1)>= i - arr[i]则a是合法的,通过二分查找找到最大的a
2.问题转化为在区间[x,n - y]之间找到可删除个数大于等于x的个数,进一步变为[1,n - y]之间可删除个数大于等于x的个数,因为在[1,x)之间不可能有点可删除个数 >= x,查询对x从大到小排序,可删除个数数组对可删除个数a进行排序,离线查询[1,n - y]可删除个数>= x的个数

#include
#define debug(x) cout<<#x<<" is "<
#define DBG 0 
#define pb push_back
using namespace std;
const int N = 3e5 + 5;
int n,q;
int a[N],b[N],c[N];
int low_bit(int x){
     
	return x&(-x);
}
void add(int i){
     
	if(i == 0){
     
		b[0]++;//最多删除0个情况特判
		return ;
		
	}
	while(i <= n){
     
		b[i]++;
		i += low_bit(i);
	}
}
int query(int i){
     //返回sum(0~i)个个数
	if(i < 0)return 0;
	int ans = 0;
	while(i > 0){
     
		ans += b[i];
		i -= low_bit(i);
	}
	ans += b[0];
	return ans;
}
void add0(int i){
     
	while(i <= n){
     
		c[i]++;
		i += low_bit(i);
	}
}
int query0(int i){
     
	int ans = 0;
	while(i > 0){
     
		ans += c[i];
		i -= low_bit(i);
	}
	return ans;
}
int binary_search(int l,int r,int i){
     
	int ans = -1;
	while(l <= r){
     
		
		int mid = (l + r)>>1;
		if(query(i) - query(mid - 1) >= i - a[i]){
     
			ans = mid;
			l = mid + 1;
		}else r = mid - 1;
	}
	return ans;
}

struct point{
     
	int x,y,index;
	int ans;
	point(int _x,int _y,int _index):x(_x),y(_y),index(_index){
     
		ans = 0;
	} 
	
};
struct node{
     
	int x,index;
	node(int _x,int _index):x(_x),index(_index){
     
	}
	bool operator<(node &n){
     
		return x > n.x;
	}
};
bool cmp1(point&p1,point &p2){
     
	return p1.x > p2.x;
}
bool cmp2(point&p1,point &p2){
     
	return p1.index < p2.index;
}
vector<point> v;
vector<node> vv;
int main(){
     
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>q;
	for(int i = 1;i <= n;i++)cin>>a[i];

	for(int i = 1;i <= n;i++){
     
		if(i - a[i] < 0)vv.pb(node(-1,i));
		else if(i - a[i] == 0){
     
			vv.pb(node(i - 1,i));
			add(i - 1);
		}else{
     
			int t = binary_search(0,i - 1,i);
			vv.pb(node(t,i));
			if(t != -1)add(t);
		}	
	}
	sort(vv.begin(),vv.end());
	for(int i = 1;i <= q;i++){
     
		int x,y;
		cin>>x>>y;
		v.pb(point(x,y,i));
	}
	sort(v.begin(),v.end(),cmp1);
	int j = 0;
	for(int i = 0;i < v.size();i++){
     
		int x = v[i].x,y = v[i].y;
		while(vv[j].x >= x&&j < vv.size())add0(vv[j].index),j++;
		v[i].ans = query0(n - y);
	}
	sort(v.begin(),v.end(),cmp2);
	for(int i = 0;i < v.size();i++){
     
		cout<<v[i].ans<<endl;
	}
	return 0;
}

PS:主席树似乎更快更好,但是还没学,留个坑。

你可能感兴趣的:(树状数组)