CodeForces 1514D : Cut and Stick 主席树

传送门

题目描述

现在给你一个序列$a_i$，我们定义一个好的序列为序列中每一种数字出现的次数都不超过$\frac{n}{2}$，$n$为序列总长度，现在给你一个区间$l,r$，问你最少将这个区间分割成几部分，可以使每个子区间都是一个好区间

分析

首先分析一下一个区间内只会有一个数出现的次数大于$\frac{n}{2}$
如果这个区间没有任何一个数出现的次数大于$\frac{n}{2}$，返回1
如果有，假设出现次数为$x$，那么剩下的数的数量为$n-x$，所以我可以从$x$中挑出$n-x+1$个数和其他的数合并，剩下的数只能单独划开一个区间，所以最后需要的区间数量为$x+1-(n-x+1)=2\ast x-n$
我们可以在主席树上二分找到这个出现次数最大的数，然后计算正确答案

代码

#pragma GCC optimize(3)
#include <bits/stdc++.h>
#define debug(x) cout<<#x<<":"<<x<<endl;
#define dl(x) printf("%lld\n",x);
#define di(x) printf("%d\n",x);
#define _CRT_SECURE_NO_WARNINGS
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef vector<int> VI;
const int INF = 0x3f3f3f3f;
const int N = 5e5 + 10;
const ll mod= 1000000007;
const double eps = 1e-9;
const double PI = acos(-1);
template<typename T>inline void read(T &a){char c=getchar();T x=0,f=1;while(!isdigit(c)){if(c=='-')f=-1;c=getchar();}
while(isdigit(c)){x=(x<<1)+(x<<3)+c-'0';c=getchar();}a=f*x;}
int gcd(int a,int b){return (b>0)?gcd(b,a%b):a;}
int root[N];
struct Node{
    int l,r;
    int cnt;
}tr[N * 66];
int a[N];
int idx,n,m;

int insert(int u,int l,int r,int x){
    int q = ++idx;
    tr[q] = tr[u];
    tr[q].cnt++;
    if(l == r) return q;
    int mid = l + r >> 1;
    if(x <= mid) tr[q].l = insert(tr[u].l,l,mid,x);
    else tr[q].r = insert(tr[u].r,mid + 1,r,x);
    return q;
}

int query(int p,int q,int l,int r,int k){
    int limet = k / 2 + (k % 2);
    if(l == r){
        int cnt = tr[q].cnt - tr[p].cnt;
        if(cnt <= limet) return 1;
        else return 2 * cnt - k;
    }
    int cntl = tr[tr[q].l].cnt - tr[tr[p].l].cnt;
    int cntr = tr[tr[q].r].cnt - tr[tr[p].r].cnt;
    int mid = l + r >> 1;
    if(cntl <= limet && cntr <= limet) return 1;
    if(cntl <= limet && cntr > limet) return query(tr[p].r,tr[q].r,mid + 1,r,k);
    if(cntl > limet && cntr <= limet) return query(tr[p].l,tr[q].l,l,mid,k);
    return max(query(tr[p].r,tr[q].r,mid + 1,r,k),query(tr[p].l,tr[q].l,l,mid,k));
}

int main(){
    read(n),read(m);
    for(int i = 1;i <= n;i++) {
        read(a[i]);
        root[i] = insert(root[i - 1],1,300000,a[i]);
    }
    while(m--){
        int l,r;
        read(l),read(r);
        di(query(root[l - 1],root[r],1,300000,r - l + 1));
    }
    return 0;
}

/**
*　　┏┓　　　┏┓+ +
*　┏┛┻━━━┛┻┓ + +
*　┃　　　　　　　┃
*　┃　　　━　　　┃ ++ + + +
*  ████━████+
*  ◥██◤　◥██◤ +
*　┃　　　┻　　　┃
*　┃　　　　　　　┃ + +
*　┗━┓　　　┏━┛
*　　　┃　　　┃ + + + +Code is far away from 　
*　　　┃　　　┃ + bug with the animal protecting
*　　　┃　 　 ┗━━━┓ 神兽保佑,代码无bug　
*　　　┃ 　　　　　　 ┣┓
*　　  ┃ 　　　　　 　┏┛
*　    ┗┓┓┏━┳┓┏┛ + + + +
*　　　　┃┫┫　┃┫┫
*　　　　┗┻┛　┗┻┛+ + + +
*/