P8787 [蓝桥杯 2022 省 B] 砍竹子--贪心

[蓝桥杯 2022 省 B] 砍竹子 - 洛谷

思路：首先我们读完题，很容易想到暴力，就是枚举所有序列，然后判断是否有连续相同的数，但是这样肯定会爆掉。

小tip：一定要审题，题目说的是连续的一段相同的竹子。还有范围是longlong

那么我们如何优化呢？我们可以预处理出所有所有竹子砍的次数，最大的为mx，那么我们会发现，mx次之后，所有的竹子一定都可以变成1.

代码：

#include <bits/stdc++.h>
#define int long long //(有超时风险)
#define PII pair<int,int>
#define endl '\n'
#define LL __int128

using namespace std;

const int N=2e5+10,M=1e3+10,mod=998244353,INF=0x3f3f3f3f;

int h[N],cnt[N],c[N],pre[N];

signed main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);

    int n;cin>>n;

    //记录mx次，和每个竹子砍的次数
    int mx=0;
    for(int i=1;i<=n;i++)
    {
        cin>>h[i];

        int x=h[i];
        while(x!=1)
        {
            x=sqrt(x/2+1);
            cnt[i]++;
        }
        mx=max(mx,cnt[i]);
    }

    //枚举mx，枚举竹子。判断答案。
    int ans=0;
    for(int i=mx;i>=1;i--)
    {
        for(int j=1;j<=n;j++)
        {
            if(cnt[j]==i)
            {
                if(h[j]!=h[j+1])
                    ans++;
                cnt[j]--;
                h[j]=sqrt(h[j]/2+1);
            }
        }
    }
    cout<<ans<<endl;
    return 0;
}