Frequent values POJ - 3368(線段樹,區間合併)
阿新 • • 發佈:2018-12-01
Frequent values POJ - 3368
題目連結
題意:一個非遞減序列,隨機詢問區間[l, r]中出現次數最多的數的出現次數;
思路:多次詢問,首先就要想一下線段樹;由題意可知數列中的數是連續的,既然是連續的就有合併的希望!!!那麼就來一發線段樹吧(RMQ也可以做,但是不會啊!!!還是蒟蒻);
首先,要用線段樹維護哪些值?
- 題目要求區間內出現次數最多的數的出現次數,那麼這一個值是必須要有的,記為maxx吧;
- 對於一個連續的區間[l, r],若它是由兩個區間[l, mid], [mid+1, r]合併過來的,那麼可以合併的前提條件是:
- a[l]==a[mid+1]:左區間和右區間的左半部分合並;
- a[mid]==a[r]:左區間的右半部分和右區間合併;
- a[l]==a[r]:左區間和右區間合併;
- a[mid]==a[mid+1]:左區間的右半部分和右區間的左半部分合並;
那麼 我們就需要記錄lson, rson分別用來表示該區間最左邊的數的出現次數和最右邊數的出現次數;
然後就直接建樹,然後查詢;
對於查詢同樣要和合並一樣分情況;
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <iostream>
using namespace std;
const int maxn=1e5+10;
struct node{
int l, r, lson, rson, maxx;
}tr[maxn<<2];
int a[maxn];
void pushup(int m){
//先繼承左右孩子的最大值;
tr[m].lson=tr[m<<1].lson;
tr[m].rson=tr[m<<1|1].rson;
tr[m].maxx=max(tr [m<<1].maxx, tr[m<<1|1].maxx);
//然後就行合併;
int len=0, lson=0, rson=0, maxx=0;
if(a[tr[m<<1].l]==a[tr[m<<1|1].r]){//a[l]==a[l]
len=tr[m<<1].lson+tr[m<<1|1].rson;
}
if(a[tr[m<<1].l]==a[tr[m<<1|1].l]){//a[l]==a[mid+1]
lson=tr[m<<1].lson+tr[m<<1|1].lson;
}
if(a[tr[m<<1].r]==a[tr[m<<1|1].r]){//a[mid]==a[r]
rson=tr[m<<1|1].rson+tr[m<<1].rson;
}
if(a[tr[m<<1].r]==a[tr[m<<1|1].l]){//a[mid]==a[mid+1]
maxx=tr[m<<1].rson+tr[m<<1|1].lson;
}
//維護最值;
tr[m].lson=max(tr[m].lson, max(len, lson));
tr[m].rson=max(tr[m].rson, max(len, rson));
tr[m].maxx=max(tr[m].maxx, max(maxx, len));
tr[m].maxx=max(tr[m].maxx, max(tr[m].lson, tr[m].rson));
//printf("m:%d l:%d r:%d lson:%d maxx:%d rson:%d\n", m, tr[m].l, tr[m].r, tr[m].lson, tr[m].maxx, tr[m].rson);
}
void build(int m, int l, int r){
tr[m].l=l;
tr[m].r=r;
if(l==r){
tr[m].lson=tr[m].rson=tr[m].maxx=1;
return;
}
int mid=(l+r)>>1;
build(m<<1, l, mid);
build(m<<1|1, mid+1, r);
pushup(m);
}
//得到區間[l, r]中最左邊數的出現次數;
int query_lson(int m, int l, int r){
//printf("query_lson: m:%d tr[m].l:%d tr[m].r:%d l:%d r:%d\n", m, tr[m].l, tr[m].r, l, r);
if(tr[m].l==l&&tr[m].r==r){
//cout << ".............." << endl;
return tr[m].lson;
}
int mid=(tr[m].l+tr[m].r)>>1;
if(r<=mid) return query_lson(m<<1, l, r);
else if(l>mid) return query_lson(m<<1|1, l, r);
else{
if(a[l]==a[mid+1]) return mid-l+1+query_lson(m<<1|1, mid+1, r);
else return query_lson(m<<1, l, mid);
}
}
//得到區間[l, r]中最右邊數的出現次數;
int query_rson(int m, int l, int r){
//printf("query_son: m:%d tr[m].l:%d tr[m].r:%d l:%d r:%d\n", m, tr[m].l, tr[m].r, l, r);
if(tr[m].l==l&&tr[m].r==r){
//cout << ".............." << endl;
return tr[m].rson;
}
int mid=(tr[m].l+tr[m].r)>>1;
if(r<=mid) return query_rson(m<<1, l, r);
else if(l>mid) return query_rson(m<<1|1, l, r);
else{
if(a[r]==a[mid]) return r-mid+query_rson(m<<1, l, mid);
else return query_rson(m<<1|1, mid+1, r);
}
}
//得到區間[l, r]中最大出現次數;
int query(int m, int l, int r){
//printf("query: m:%d tr[m].l:%d tr[m].r:%d l:%d r:%d\n", m, tr[m].l, tr[m].r, l, r);
if(tr[m].l==l&&tr[m].r==r){
//cout << ".............." << endl;
return tr[m].maxx;
}
int mid=(tr[m].l+tr[m].r)>>1;
if(r<=mid) return query(m<<1, l, r);
else if(l>mid) return query(m<<1|1, l, r);
else{
int lson=0, rson=0, mson=0, max1=0, max2=0;
if(a[l]==a[r]){
return r-l+1;
}
//查詢時的合併;
if(a[l]==a[mid+1]) lson=mid-l+1+query_lson(m<<1|1, mid+1, r);
if(a[mid]==a[r]) rson=r-mid+query_rson(m<<1, l, mid);
if(a[mid]==a[mid+1]) mson=query_rson(m<<1, l, mid)+query_lson(m<<1|1, mid+1, r);
max1=query(m<<1, l, mid), max2=query(m<<1|1, mid+1, r);
return max(lson, max(rson, max(mson, max(max1, max2))));
}
}
int main(){
int n, q;
while(scanf("%d", &n), n){
scanf("%d", &q);
for(int i=1; i<=n; i++){
scanf("%d", &a[i]);
}
build(1, 1, n);
while(q--){
int i, j;
scanf("%d%d", &i, &j);
printf("%d\n", query(1, i, j));
}
}
return 0;
}