[JSOI2018]D1T2 iiidx [线段树+二分]

显然题意可以抽象成:一棵树,要对树上的每个点标上给定的权值,满足每个点上的权值都 ≤\leq 子树内点的权值,并使这个棵树编号从小到大的权值字典序最大。
  首先可以一眼得到一个做法,将权值从大到小排序,把长度为子树大小的一段按子树编号从小到大丢给它们,递归下去得到答案。
  这种做法在 did_idi 不重复的时候是正确的,那 did_idi 要是有相同的数呢?
  有可能可以将编号 xxx 子树里一个大的权值与编号 x+1x+1x+1 的子树根的权值替换,使得 xxx 的权值依然是能取得的最大数且子树内的数都比 xxx 的权值大,同时 x+1x+1x+1 子树内的点也还能标满 ≥\geq x+1x+1x+1 权值的权值。
  那怎么做呢?先把给定权值从大到小排序,线段树上维护每个权值左边(包括本身)还能取的权值的个数 CiC_iCi
  当取好一个点 xxx 的权值时,需要给它子树内的点预留取的权值,这些权值显然在 xxx 取得权值的左边,但是我们并不知道取的是哪些权值,所以只把 xxx 取得的权值右边(包括本身)的 CiC_iCi 减去 xxx 子树大小 size[x]size[x]size[x] ,每次取一个点 yyy 的权值时,只需要在线段树上找到最大权值 valvalval 满足 valvalval 所在位置右边(包括本身)的所有 Ci≥size[y]C_i\geq size[y]Cisize[y] ,且 valvalval 尽可能靠右即可,这个在线段树上二分就能做到。
  如果一个点有父亲,求它的权值时要把它父亲为子树预留的大小去掉,需要注意的是,每个父亲预留的大小只要去掉一次,写的时候容易忽略这一点,WA了半天。

/*
 * @Author: SugarSBN
 * @Date: 2018-05-25 17:24:46
 * @Language:   C++
 * @Task:iiidx
 */
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
const int MAXN=5e6+11;
const double pi=acos(-1);
typedef long long ll;
typedef vector vi;
typedef pair pii;
typedef double dbl;
#define For(a,b,c) for (int (a)=(b);(a)<(c);(a)++)
#define foreach(iter,V) for (__typeof((V.begin()) iter=(V).begin();iter!=(V).end();iter++)
#define mp make_pair
#define fi first
#define se second
#define pb push_back
#define lch(u) u << 1
#define rch(u) u << 1 | 1
int n, a[MAXN];
int cnt[MAXN], siz[MAXN], fa[MAXN], ans[MAXN];
double k;
bool cmp(int a,int b){
    return a > b;
}
inline void init(){
    scanf("%d%lf", &n, &k);
    For(i, 1, n + 1)    scanf("%d", a + i);
    sort(a + 1, a + n + 1, cmp);
    for (int i = n - 1; i > 0;i--)
        if (a[i]==a[i+1])
            cnt[i] = cnt[i + 1] + 1;
        else
            cnt[i] = 0;
    For(i, 1, n+1){
        fa[i] = floor(i / k);
        siz[i] = 1;
    }
    for (int i = n; i;i--)
        siz[fa[i]] += siz[i];
}
struct node{
    int mn, delta;
    int l, r;
} T[MAXN << 1];
void pushup(int o){
    T[o].mn = min(T[lch(o)].mn, T[rch(o)].mn);
}
void pushdown(int o){
    T[lch(o)].mn += T[o].delta;
    T[rch(o)].mn += T[o].delta;
    T[lch(o)].delta += T[o].delta;
    T[rch(o)].delta += T[o].delta;
    T[o].delta = 0;
}
void build(int o,int l,int r){
    T[o].l = l;
    T[o].r = r;
    T[o].mn = l;
    if (l==r)
        return;
    int mid = (l + r) >> 1;
    build(lch(o), l, mid);
    build(rch(o), mid+1, r);
    pushup(o);
}
void update(int o,int l,int r,int delta){
    if (l<=T[o].l&&T[o].r<=r){
        T[o].mn += delta;
        T[o].delta += delta;
        return;
    }
    if (l>T[o].r||T[o].r> 1;
    if (l<=mid)
        update(lch(o),l,r,delta);
    if (r>mid)
        update(rch(o), l, r, delta);
    pushup(o);
}
int query(int o,int k){
    if (T[o].l==T[o].r)
        return T[o].mn >= k ? T[o].l : T[o].l + 1;
    pushdown(o);
    int mid = (T[o].l + T[o].r) >> 1;
    if (k<=T[rch(o)].mn)
        return query(lch(o), k);
    else
        return query(rch(o), k);
}
inline void solve(){
    For(i, 1, n + 1){
        if (fa[i]&&fa[i]!=fa[i-1])
            update(1, ans[fa[i]], n, siz[fa[i]] - 1);
        int x = query(1, siz[i]);
        x += cnt[x];
        cnt[x]++;
        x -= (cnt[x] - 1);
        ans[i] = x;
        update( 1, x, n, -siz[i]);
    }
}
int main(){
    //freopen("iiidx.in", "r", stdin);
    init();
    build(1, 1, n);
    solve();
    For(i, 1, n + 1)
        printf("%d ", a[ans[i]]);
    return 0;
}

你可能感兴趣的:([JSOI2018]D1T2 iiidx [线段树+二分])