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

本文介绍了一种利用线段树优化树形DP的问题解决策略,通过将给定权值排序并维护每个权值可选取的数量,实现了对树上每个节点标记权值的过程,确保了每个节点的权值不超过其子树中任意节点的权值,并最大化了字典序。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

显然题意可以抽象成:一棵树,要对树上的每个点标上给定的权值,满足每个点上的权值都≤\leq子树内点的权值,并使这个棵树编号从小到大的权值字典序最大。
  首先可以一眼得到一个做法,将权值从大到小排序,把长度为子树大小的一段按子树编号从小到大丢给它们,递归下去得到答案。
  这种做法在did_idi不重复的时候是正确的,那did_idi要是有相同的数呢?
  有可能可以将编号xxx子树里一个大的权值与编号x+1x+1x+1的子树根的权值替换,使得xxx的权值依然是能取得的最大数且子树内的数都比xxx的权值大,同时x+1x+1x+1子树内的点也还能标满≥\geqx+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<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<queue>
#include<stack>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<cmath>
#include<sstream>
#include<fstream>
using namespace std;
const int MAXN=5e6+11;
const double pi=acos(-1);
typedef long long ll;
typedef vector<int> vi;
typedef pair<int,int> 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<l)
        return;
    pushdown(o);
    int mid = (l + 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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值