【贪心+线段树】Codeforces 557C Arthur and Table

本文介绍了一种使用线段树数据结构解决桌脚优化问题的方法。通过暴力枚举最长桌脚并调整其他桌脚长度来最小化总成本,采用线段树维护长度分布,实现高效查询和更新。

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

题面在这里

其实很简单……

暴枚最长桌脚的长度 l ,然后长度比l长的桌脚全部都要砍掉

长度比 l 短的桌脚选择代价前k小的砍掉

用线段树维护……

示例程序 :

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long ll;
inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int red(){
    int res=0,f=1;char ch=nc();
    while (ch<'0'||'9'<ch) {if (ch=='-') f=-f;ch=nc();}
    while ('0'<=ch&&ch<='9') res=res*10+ch-48,ch=nc();
    return res*f;
}

const int maxn=100005;
int n,m,l[maxn];
ll ans=1ll<<62,sum=0;
vector<int> a[maxn];
struct node{
    node *l,*r;
    int L,R,s;ll w;
    node () {}
    node (int _L,int _R):L(_L),R(_R),s(0),w(0) {}
    void pushup() {s=l->s+r->s; w=l->w+r->w;}
}nil,bace[maxn<<1];
typedef node* P_node;
P_node null,len,Rot;
void init(){
    nil=node(0,0);null=&nil;
    null->l=null->r=null;len=bace;
}
P_node newnode(int L,int R){
    *len=node(L,R);
    len->l=len->r=null;
    return len++;
}
P_node build(int L,int R){
    P_node x=newnode(L,R);
    if (L==R) return x;
    int mid=L+R>>1;
    x->l=build(L,mid); x->r=build(mid+1,R);
    return x;
}
void insert(P_node x,int k,int wi){
    if (x->L==x->R) {x->s+=wi;x->w=(ll)x->s*x->L;return;}
    int mid=x->L+x->R>>1;
    if (k<=mid) insert(x->l,k,wi);else insert(x->r,k,wi);
    x->pushup();
}
ll query(P_node x,int k){
    if (k> x->s) return 1ll<<62;
    if (x->L==x->R) return (ll)x->L*k;
    if (k<= x->l->s) return query(x->l,k);
    return query(x->r,k- x->l->s)+ x->l->w;
}
int main(){
    n=red();m=100000;
    for (int i=1;i<=n;i++) l[i]=red();
    for (int i=1,x;i<=n;i++) x=red(),sum+=x,a[l[i]].push_back(x);
    for (int i=1;i<=m;i++){
        int Max=0,cd=0;
        for (int j : a[i])
         if (j>=Max) cd=Max,Max=j;else
         if (j>cd) cd=j;
        if (Max) ans=min(ans,sum-Max);
        if (cd) ans=min(ans,sum-Max-cd);
    }
    init();Rot=build(1,m);
    for (int i=1,all=0;i<=m;i++){
        for (int j: a[i]) sum-=j,all++;
        if (all==a[i].size()&&a[i].size()>2) ans=min(ans,sum);
        if (a[i].size()>1&&all>=a[i].size()*2-1) ans=min(ans,sum+query(Rot,all-a[i].size()*2+1)) ;
        for (int j: a[i]) insert(Rot,j,1);
    }
    printf("%I64d",ans);
    return 0; 
}
### 解题思路 #### 问题描述 Codeforces 1678C - Tokitsukaze and Strange Inequality 是一道关于排列组合与前缀和的应用问题。给定一个长度为 \( n \) 的排列数组 \( p \),需要统计满足条件 \( a < b < c < d \) 并且 \( p_a < p_c \) 同时 \( p_b > p_d \) 的四元组数量。 --- #### 核心思想 由于数据规模较小 (\( n \leq 5000 \)),可以直接通过枚举的方式解决问题。为了降低时间复杂度,引入 **前缀和** 技术来加速计算过程[^3]。 具体来说: - 枚举变量 \( a \) 和 \( c \),固定它们之后,目标是快速找到符合条件的 \( b \) 和 \( d \)。 - 使用预处理好的前缀和数组 `num` 来高效查询某个范围内满足特定关系的数量。 - 定义辅助数组 `sum` 表示对于固定的区间范围内的某些约束条件下的累积计数结果。 --- #### 实现细节 ##### 步骤一:构建前缀和数组 `num` 定义二维数组 `num[i][j]`,其中 `num[i][j]` 表示在序列的前 \( i \) 项中,有多少个元素大于 \( j \)。 该数组可以通过如下方式初始化: ```python n = len(p) max_val = max(p) # 初始化 num 数组 num = [[0] * (max_val + 2) for _ in range(n + 1)] for i in range(1, n + 1): for j in range(max_val + 1, -1, -1): # 反向遍历以保持正确性 if p[i - 1] > j: num[i][j] = num[i - 1][j] + 1 else: num[i][j] = num[i - 1][j] ``` 上述代码的时间复杂度为 \( O(n \cdot m) \),其中 \( m \) 是数组中的最大值。 --- ##### 步骤二:定义并填充辅助数组 `sum` 定义另一个二维数组 `sum[i][j]`,它表示当 \( a=i \), \( c=j \) 时,在区间 \([a+1, c-1]\) 中满足 \( p[b] > p[d] \) 的总贡献次数。 利用动态规划的思想逐步更新此数组: ```python sum_ = [[0] * (n + 1) for _ in range(n + 1)] bucket = [0] * (max_val + 1) for l in range(n - 1, 0, -1): bucket[p[l]] += 1 for r in range(l + 2, n + 1): sum_[l][r] = sum_[l][r - 1] + (num[r - 1][p[r - 1]] - num[l][p[r - 1]]) ``` 这里的关键在于如何有效累加当前区间的合法贡献,并借助之前已经计算的结果减少重复运算。 --- ##### 步骤三:枚举所有可能的 \( a \) 和 \( c \) 最后一步是对所有的 \( a \) 和 \( c \) 进行双重循环,并将对应位置上的 `sum[a][c]` 加入最终答案中: ```python result = 0 for a in range(1, n - 2): for c in range(a + 2, n): result += sum_[a][c] print(result) ``` 整个算法的核心部分即完成以上三个阶段的操作即可实现高效的解决方案。 --- ### 总结 本题主要考察的是对多重嵌套结构的有效简化以及合理运用前缀和技巧的能力。通过巧妙设计的数据结构能够显著提升程序运行效率至可接受水平。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值