3993. 石子游戏 模拟 前缀和

题目

题解思路

一开始用全部多的数来二分搜答案，后面wa了才发现是思路错了。
因为不一定前面的每次我们都能取到k个数，我们只能通过模拟来让每一步合法。
模拟控制 h，每次取的数要尽量靠近K不能大。
将将石子堆的数放入桶中，取最大的和最小的作为上下界。
求出h数组前缀和这样我们就能O1的求出某一部分的数的个数了。
一层一层的控制取出的总石子K的个数当大于时就加一次个数

这样就能贪心出最小次数

这里可能有点难理解 ， 因为j时一层一层往下的所以直接加就是石子数，h[ma] - h[j-1]就是大于J的石子的总数 因为是一层一层减的所以次数h[ma] - h[j-1]的所有石子就只有J+1个

    for (int i = ma ; i > mi ; )
    {
        int j = i , sum = 0 ; 
        while( j > mi && (sum + h[ma] - h[j-1]) <= k )
        {
            sum += h[ma] - h[j-1];
            j--;
        }
        ans ++; 
        i = j ;
    }

AC代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>
#include <map>
#include <string>
using namespace std;

const  int  INF =  0x3f3f3f3f ;
const  int  N = 200100 ;

int h[N];
int n , k , mi = INF , ma = -1 ;


int main ()
{
    ios::sync_with_stdio(false);
    
    cin >> n >> k  ;
    
    for (int i = 1 ; i <= n ; i++ )
    {
        int tt ;
        cin>>tt;
        h[tt]++;
        mi = min( mi , tt );
        ma = max (ma , tt );
    }
    for (int i = 1 ; i <= ma ; i++ )  h[i] += h[i-1] ;
    int ans = 0 ; 
    for (int i = ma ; i > mi ; )
    {
        int j = i , sum = 0 ; 
        while( j > mi && (sum + h[ma] - h[j-1]) <= k )
        {
            sum += h[ma] - h[j-1];
            j--;
        }
        ans ++; 
        i = j ;
    }
    cout << ans << "\n" ;
    
    return 0 ;

}