[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<bits/stdc++.h>
#define debug(x) cout<<#x<<" is "<<x<<endl
#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:主席树似乎更快更好,但是还没学,留个坑。

猜你喜欢

转载自blog.csdn.net/qq_20252251/article/details/108539693