[CF522D]Closest Equals——线段树+离线技巧+思维

最新推荐文章于 2024-07-03 14:52:54 发布

Echo_ac

最新推荐文章于 2024-07-03 14:52:54 发布

阅读量247

点赞数

分类专栏：数据结构 # 思维 # ACM-线段树

本文链接：https://blog.csdn.net/qq_45769627/article/details/123394444

版权

线段树离线处理区间查询数据结构算法

关键词由CSDN通过智能技术生成

# 思维同时被 3 个专栏收录

34 篇文章 0 订阅

订阅专栏

数据结构

9 篇文章 0 订阅

订阅专栏

ACM-线段树

4 篇文章 0 订阅

订阅专栏

题目传送门

思路

应该能想到我们需要求出每一个 $a_i$ 上一次出现的位置 $p r e [i]$ , 对于每一个询问区间 $[l, r]$ 我们需要求出 $i - p r e [i]$ 的最小值 $\geq l)$ 。
线段树维护最小值即可
问题1：我们可以将每个 $p r e [i]$ 位置的值换成 $i - p r e [i]$ ，但是每次在线暴力置换一定会超时
我们将询问离线下来，遍历 $1$ ~ $n$ , 对于每一个 $i$ ，我们执行修改操作，修改 $p r e [i]$ 位置的值为 $i - p r e [i]$ ，这样便不会超时。
问题2：如果只是按照上述操作后再来查询，可能会出现查询时的 $i > r$ ，就是 $p r e [i]$ 再区间 $[l, r]$ 内，但是 $i$ 不一定在。
我们应该这样做，在遍历 1~n 的时候，将询问按照右端点从小到大排序，当 $i = = r$ 的时候，我们就去查询 $[l, r]$ ，这样便能正确解决此问题。

AC代码

#include<bits/stdc++.h>
#define rep(i,x,y) for(int i=x; i<=y; ++i)
#define per(i,x,y) for(int i=x; i>=y; --i)
#define pushk push_back
#define popk pop_back
#define mem(a,b) memset(a,b,sizeof a)
#define PII pair<int,int>
#define ll long long
#define lp p<<1
#define rp p<<1|1
#define INF 0x3f3f3f3f
using namespace std;
const int N = 5e5+9;
int a[N],ans[N];
struct node{
	int l,r,mi;
}tr[N<<2];
struct ASK{
	int l,r,id;
	bool operator<(const ASK &t) const{
		return r<t.r;
	}
}q[N];
unordered_map<int,int> mp;
void build(int p,int l,int r){
	tr[p].l=l,tr[p].r=r;
	if(l==r) {
	tr[p].mi = INF;
		return ;
	}
	int mid = (l+r)>>1;
	build(lp,l,mid);
	build(rp,mid+1,r);
	tr[p].mi = min (tr[lp].mi, tr[rp].mi); 
}
void modify(int p,int x,int c){
	if(tr[p].l==tr[p].r) {
		tr[p].mi=c;
		return ;
	}
	int mid = (tr[p].l + tr[p].r)>>1;
	if(x<=mid) modify(lp,x,c);
	else modify(rp,x,c);
	tr[p].mi = min (tr[lp].mi, tr[rp].mi); 
}
int ask(int p,int l,int r){
	if(l<=tr[p].l && tr[p].r<=r) return tr[p].mi;
	int mid = (tr[p].l + tr[p].r)>>1;
	int res=INF;
	if(l<=mid) res=min(res,ask(lp,l,r));
	if(r>mid) res=min(res,ask(rp,l,r));
	tr[p].mi = min (tr[lp].mi, tr[rp].mi); 
	return res;
}
int main() {	

	int n,m;
	cin>>n>>m;
	rep(i,1,n)  scanf("%d",a+i);
	rep(i,1,m){
		scanf("%d %d",&q[i].l,&q[i].r); 
		q[i].id=i;
		ans[i]=INF;
	}
	build(1,1,n);
	sort(q+1,q+1+m);
	int idx=1;
	rep(i,1,n){
		if(mp[a[i]]) modify(1,mp[a[i]],i-mp[a[i]]);
		while(idx<=m && q[idx].r==i){			
			ans[q[idx].id] = ask(1,q[idx].l,q[idx].r);
			idx++; 
		}
		mp[a[i]]=i;
	}
	rep(i,1,m) printf("%d\n",ans[i]==INF?-1:ans[i]);
	return 0;
}