[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<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;
}

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
根据引用[1],dp[u][j]表示在u子树中选取恰好j个人时能获得的最大价值。而根据引用,该问题的时间复杂度为O(log2​104×nm)。 对于洛谷P2143 [JSOI2010] 巨额奖金问题,我们可以使用动态规划来解决。具体步骤如下: 1. 首先,我们需要构建一棵树来表示员工之间的关系。树的根节点表示公司的总经理,其他节点表示员工。每个节点都有一个权值,表示该员工的奖金金额。 2. 接下来,我们可以使用动态规划来计算每个节点的dp值。对于每个节点u,我们可以考虑两种情况: - 如果选择节点u,则dp[u][j] = dp[v][j-1] + value[u],其中v是u的子节点,value[u]表示节点u的奖金金额。 - 如果不选择节点u,则dp[u][j] = max(dp[v][j]),其中v是u的子节点。 3. 最后,我们可以通过遍历树的所有节点,计算出dp[u][j]的最大值,即为所求的巨额奖金。 下面是一个示例代码,演示了如何使用动态规划来解决洛谷P2143 [JSOI2010] 巨额奖金问题: ```python # 构建树的数据结构 class Node: def __init__(self, value): self.value = value self.children = [] # 动态规划求解最大奖金 def max_bonus(root, j): dp = [[0] * (j+1) for _ in range(len(root)+1)] def dfs(node): if not node: return for child in node.children: dfs(child) for k in range(j, 0, -1): dp[node.value][k] = max(dp[node.value][k], dp[node.value][k-1] + node.value) for child in node.children: for k in range(j, 0, -1): for l in range(k-1, -1, -1): dp[node.value][k] = max(dp[node.value][k], dp[node.value][k-l-1] + dp[child.value][l]) dfs(root) return dp[root.value][j] # 构建树 root = Node(1) root.children.append(Node(2)) root.children.append(Node(3)) root.children[0].children.append(Node(4)) root.children[0].children.append(Node(5)) root.children[1].children.append(Node(6)) # 求解最大奖金 j = 3 max_bonus_value = max_bonus(root, j) print("最大奖金为:", max_bonus_value) ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值