[CF1065C] Make It Equal 解题记录

[CF1065C] Make It Equal 解题记录

---

题意简述

有一个由 \(n\) 座塔组成的建筑，编号为 \(i\) 的塔由 \(h_i\) 个立方体组成。
现在可以对其进行切割操作：选定一个整数 \(H\)，把所有高度大于 \(H\) 的塔都切成 \(H\) 高度，消耗的能量值为切割的高度。并且消耗的能量不能大于给定的整数 \(k\)。
求最少切割次数。

---

题目分析

对于样例

5 5
3 1 2 2 4

可以得到这个图：

我们把它顺时针旋转 \(90°\)：

这时问题就从原来的横着切割变成了纵向切割，每个塔的高度 \(h_i\) 就是区间 \([1,h_i]\)，用差分区间修改，然后前缀和还原数组 \(s\)，切割位置 \(i\) 消耗的能量就是 \(s_i\)。
模拟整个过程即可。

---

AC Code

#include<bits/stdc++.h>
#define arrout(a,n) rep(i,1,n)std::cout<<a[i]<<" "
#define arrin(a,n) rep(i,1,n)std::cin>>a[i]
#define rep(i,x,n) for(int i=x;i<=n;i++)
#define dep(i,x,n) for(int i=x;i>=n;i--)
#define erg(i,x) for(int i=head[x];i;i=e[i].nex)
#define dbg(x) std::cout<<#x<<":"<<x<<" "
#define mem(a,x) memset(a,x,sizeof a)
#define all(x) x.begin(),x.end()
#define arrall(a,n) a+1,a+1+n
#define PII std::pair<int,int>
#define m_p std::make_pair
#define u_b upper_bound
#define l_b lower_bound
#define p_b push_back
#define CD const double
#define CI const int
#define int long long
#define il inline
#define ss second
#define ff first
#define itn int
CI N=2e5+5;
int n,k,sum,lim,ans,a[N],s[N];
signed main() {
    std::cin>>n>>k;
    sum=k+1;
    rep(i,1,n) {
        int x;
        std::cin>>x;
        a[1]++,a[x+1]--;//区间[1,x]+1
        lim=std::max(lim,x);//上限
    }
    rep(i,1,lim) {//还原数组
        s[i]=s[i-1]+a[i];
    }
    dep(i,lim,1) {//模拟，从高到低切割
        if(s[i]==n) {
            break;
        }
        if(sum+s[i]<=k) {
            sum+=s[i];
        } else {
            sum=s[i];
            ans++;
        }
    }
    std::cout<<ans;
    return 0;
}