傳送門

主席樹大好題……這道題讓主席樹不僅停留在了區間第k大上，而是讓它能執行像線段樹一樣的操作。

首先我們先說點套路的事。求中位數有一個二分法，就是每次二分答案，把大于等于當前二分的數設為1，小于的設為-1，之后我們只要看和是否大于0就能判斷限制二分的值是大是小。然后雖然區間是不確定的，但是我們要求最大的中位數，只要取中間固定的一段+左區間最大后綴+右區間最大前綴判斷即可。

不過這樣的話每次暴力的時候，查詢是(O(nlogn))的，再套上查詢次數會超時。如何優化一下呢？如果我們用線段樹對于每一個二分的值，都維護一下它當前的值和最大前后綴，這樣的話我們就只需要(O(log^2n))的復雜度來完成一次查詢。但是這樣會(MLE)。

然后在每兩棵樹之間，其實只有(i-1)發生了變化，從1變成了-1.這樣的話我們就可以把這些線段樹建成主席樹，用主席樹來實現線段樹一樣的操作就好了。注意這道題是把以當前這個數的值為根，以位置為值域，注意訪問區間不要開小（我就因為這個大數據(MLE)），最開始建樹的地方比較重要，我們需要開一個(vector)來把數的值一樣，多次出現的數的位置都存起來，然后按照值的大小按順序修改即可。

看一下代碼。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<cmath>
#include<set>
#include<vector>
#include<map>
#include<queue>
#define rep(i,a,n) for(int i = a;i <= n;i++)
#define per(i,n,a) for(int i = n;i >= a;i--)
#define enter putchar('
')
#define fr friend inline
#define y1 poj
#define mp make_pair
#define pr pair<int,int>
#define fi first
#define sc second
#define pb push_back
#define I puts("Oops")

using namespace std;
typedef long long ll;
const int M = 20005;
const int N = 1000005;
const int INF = 1000000009;
const double eps = 1e-7;

int read()
{
    int ans = 0,op = 1;char ch = getchar();
    while(ch < '0' || ch > '9') {if(ch == '-') op = -1;ch = getchar();}
    while(ch >= '0' && ch <= '9') ans = ans * 10 + ch - '0',ch = getchar();
    return ans * op;
}

struct node
{
   int ls,rs,v,lv,rv;
}t[N<<2];

int n,Q,a[M],b[M],root[M],q[6],x,y,z,e,tot,la,L,R,g,idx;
vector <int> V[M];

void update(int p)
{
   t[p].v = t[t[p].ls].v + t[t[p].rs].v;
   t[p].lv = max(t[t[p].ls].lv,t[t[p].ls].v + t[t[p].rs].lv);
   t[p].rv = max(t[t[p].rs].rv,t[t[p].rs].v + t[t[p].ls].rv);
}

void build(int &p,int l,int r)
{
   if(!p) p = ++idx;
   if(l == r) {t[p].v = t[p].lv = t[p].rv = 1;return;}
   int mid = (l+r) >> 1;
   build(t[p].ls,l,mid),build(t[p].rs,mid+1,r);
   update(p);
}

void modify(int old,int &p,int l,int r,int val)
{
   p = ++idx;
   t[p] = t[old];
   if(l == r) {t[p].v = t[p].lv = t[p].rv = -1;return;}
   int mid = (l+r) >> 1;
   if(val <= mid) modify(t[old].ls,t[p].ls,l,mid,val);
   else modify(t[old].rs,t[p].rs,mid+1,r,val);
   update(p);
}

int query(int p,int l,int r,int kl,int kr)
{
   //I;
   if(kr < kl) return 0;
   if(l == kl && r == kr) return t[p].v;
   int mid = (l+r) >> 1;
   if(kr <= mid) return query(t[p].ls,l,mid,kl,kr);
   else if(kl > mid) return query(t[p].rs,mid+1,r,kl,kr);
   else return query(t[p].ls,l,mid,kl,mid) + query(t[p].rs,mid+1,r,mid+1,kr);
}

int ask(int p,int l,int r,int kl,int kr,bool flag)//1 left , 0 right
{
   //I;
   //printf("#%d %d %d %d
",l,r,kl,kr);
   if(l == kl && r == kr) return flag ? t[p].lv : t[p].rv;
   int mid = (l+r) >> 1;
   if(kr <= mid) return ask(t[p].ls,l,mid,kl,kr,flag);
   else if(kl > mid) return ask(t[p].rs,mid+1,r,kl,kr,flag);
   else
   {
      int cur1 = 0,cur2 = 0;
      cur1 = flag ? ask(t[p].ls,l,mid,kl,mid,flag) : ask(t[p].rs,mid+1,r,mid+1,kr,flag);
      if(flag) cur2 = query(t[p].ls,l,mid,kl,mid) + ask(t[p].rs,mid+1,r,mid+1,kr,flag);
      else cur2 = query(t[p].rs,mid+1,r,mid+1,kr) + ask(t[p].ls,l,mid,kl,mid,flag);
      return max(cur1,cur2);
   }
}

bool judge(int x)
{
   int cur1 = query(root[x],1,n,q[2]+1,q[3]-1);
   int cur2 = ask(root[x],1,n,q[1],q[2],0);
   int cur3 = ask(root[x],1,n,q[3],q[4],1);
   return cur1 + cur2 + cur3 >= 0;
}

int main()
{
   n = read(),build(root[0],1,n);
   rep(i,1,n) a[i] = b[i] = read();
   sort(b+1,b+1+n),tot = unique(b+1,b+1+n) - b - 1;
   rep(i,1,n) a[i] = lower_bound(b+1,b+1+tot,a[i]) - b,V[a[i]].pb(i);
   rep(i,1,tot)
   {
      root[i] = root[i-1];
      rep(j,0,(int)V[i-1].size()-1) modify(root[i],root[i],1,n,V[i-1][j]);
   }
   Q = read();
   rep(i,1,Q)
   {
      rep(j,1,4) q[j] = read(),q[j] = (q[j] + la) % n + 1;
      sort(q+1,q+5),L = 1,R = tot;
      while(L < R)
      {
	 int mid = (L+R+1) >> 1;
	 if(judge(mid)) L = mid;
	 else R = mid - 1;
      }
      la = b[L];
      printf("%d
",la);
   }
   return 0;
}

總結

以上是生活随笔為你收集整理的middle的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。