POJ 3368 區間連續個數最多的 (RMQ + 區間線段樹 2種做法)

RMQ

對於某個區間連續個數最多的數,假設這些連續的數都在這個區間,即a[l]!=a[l1]a[r]!=a[r+1]a[l]!=a[l-1]和a[r]!=a[r+1],那麼對於區間[l,r][l,r],可以先預處理出來ii這個位置連續個數numnum,然後就可以直接查詢這個區間numnum的最大值。就可以用數據結構來維護了,區間最大值(RMQ、線段樹) , 那麼對於端點外面有連續的,即a[l]==a[l1]a[r]==a[r+1]a[l]==a[l-1] 或 a[r]==a[r+1] ,那麼我們就可以直接把這段(l)(l)切掉(如果直接在這個區間求,可能會把外面的也算進去),更新到答案,後一段(r)(r)也如此。在對新的區間(沒有端點相等的)查詢區間最大值。

code

#pragma GCC optimize(2)
#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
const int man = 2e5+10;
#define IOS ios::sync_with_stdio(0)
typedef long long ll;
const ll mod = 1e9+7;

int dp[100][man];
int numl[man],numr[man],a[man];
void init(int n){
	for(int i = 1;i <= n;i++){
		dp[0][i] = numl[i] + numr[i] - 1;
	}
	int len = int(log2(n));
	for(int j = 1;j <= len;j++){
		for(int i = 1;i + (1<<j) - 1 <= n;i++){
			dp[j][i] = max(dp[j-1][i],dp[j-1][i+(1<<j-1)]);
		}
	}
}

int query(int l,int r){
	if(l>r)return 0;
	int k = int(log2(r-l+1));
	return max(dp[k][l],dp[k][r-(1<<k)+1]);
}

int main() {
	#ifndef ONLINE_JUDGE
		//freopen("in.txt", "r", stdin);
		//freopen("out.txt","w",stdout);
	#endif
	int n,m;
	while(scanf("%d",&n)!=EOF&& n){
		scanf("%d",&m);
		for(int i = 1;i <= n;i++){
			scanf("%d",&a[i]);
			numl[i] = 1;
			if(a[i]==a[i-1]){
				numl[i] += numl[i-1];
			}
		}
		for(int i = n - 1;i >= 1;i--){
			numr[i] = 1;
			if(a[i]==a[i+1])numr[i] += numr[i+1];
		}
		init(n);
		while(m--){
			int l,r;
			scanf("%d%d",&l,&r);
			int tpl = min(r - l + 1,numr[l]);
			int tpr = min(r - l + 1,numl[r]);
			int ans = 0;
			ans = max(tpl,tpr);
			l += tpl;
			r -= tpr;
			//cout << ans << endl;
			ans = max(ans,query(l,r));
			cout << ans << endl;
		}
	}
	return 0;
}

區間線段樹

思路很簡單,就是代碼有點細節,維護左邊連續相等的個數,右邊連續相等的個數,中間連續相等的個數。
還有最左邊的值和最右邊的值,
pushup往上更新合併2個區間 的時候,對於中間的一部分,有一些細節,還有查詢的時候,細節看代碼。

#pragma GCC optimize(2)
#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
const int man = 2e5+10;
#define IOS ios::sync_with_stdio(0)
typedef long long ll;
const ll mod = 1e9+7;
int a[man];
struct node{
	int l,r;
	int lva,rva;
	int mnum,lnum,rnum;
}te[man<<2];

void pushup(int rt,int l_len,int r_len){
	te[rt].lva = te[rt<<1].lva;
	te[rt].lnum = te[rt<<1].lnum;
	if(l_len==te[rt<<1].lnum&&te[rt<<1].rva==te[rt<<1|1].lva)te[rt].lnum += te[rt<<1|1].lnum;

	te[rt].rva = te[rt<<1|1].rva;
	te[rt].rnum = te[rt<<1|1].rnum;
	if(r_len==te[rt<<1|1].rnum&&te[rt<<1].rva==te[rt<<1|1].lva)te[rt].rnum += te[rt<<1].rnum;

	if(te[rt<<1].rva==te[rt<<1|1].lva)te[rt].mnum = te[rt<<1].rnum + te[rt<<1|1].lnum;
	else te[rt].mnum = 0;

	te[rt].mnum = max(te[rt].mnum,max(te[rt<<1].mnum,te[rt<<1|1].mnum));
}

void build(int l,int r,int rt){
	te[rt].l = l,te[rt].r = r;
	te[rt].lva = te[rt].rva = 0;
	te[rt].lnum = te[rt].rnum = te[rt].mnum = 0;
	if(l==r){
		te[rt].lva = te[rt].rva = a[l];
		te[rt].lnum = te[rt].rnum = te[rt].mnum = 1;
		return;
	}
	int m = l + r >>1;
	build(l,m,rt<<1);
	build(m+1,r,rt<<1|1);
	pushup(rt,m-l+1,r-m);
}

int query(int l,int r,int rt){
	if(l<=te[rt].l&&te[rt].r<=r){
		return max(te[rt].mnum,max(te[rt].lnum,te[rt].rnum));
	}
	int m = te[rt].l + te[rt].r >> 1;
	int ans = -1;
	if(r<=m){
		ans = query(l,r,rt<<1);
	}else if(l>m){
		ans = query(l,r,rt<<1|1);
	}else{
		ans = max(query(l,r,rt<<1),query(l,r,rt<<1|1));
		int tp = 0;
		if(te[rt<<1].rva==te[rt<<1|1].lva)tp = min(m-l+1,te[rt<<1].rnum) + min(r - m,te[rt<<1|1].lnum);
		ans = max(ans,tp);
	}
	return ans;
}

int main() {
	#ifndef ONLINE_JUDGE
		//freopen("in.txt", "r", stdin);
		//freopen("out.txt","w",stdout);
	#endif
	int n,m;
	while(scanf("%d",&n)!=EOF&&n){
		scanf("%d",&m);
		for(int i = 1;i <= n;i++){
			scanf("%d",&a[i]);
		}
		build(1,n,1);
		while(m--){
			int l,r;
			scanf("%d%d",&l,&r);
			printf("%d\n",query(l,r,1));
		}
	}
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章