CF1774B Coloring（贪心）

Problem - B - Codeforces

Coloring - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

有n个格子，m种颜色，每个颜色要涂ai个格子，每连续k个格子不同
将全部格子分成n/k个完整的大段，剩余的格子另说

#include <bits/stdc++.h>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#include <set>
#include <queue>
#define x first
#define int long long
#define y second
#define ios ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
using namespace std;

typedef long long LL;
typedef pair<int,int> PII;
typedef pair<char,int> PCI;
typedef pair<LL,LL> PLL;
//typedef __int128 i128;
typedef unsigned long long ULL;
const int N=2e5+10,INF = 1e9 ,mod = 1e9 + 7 ,P = 131;
const double eps = 1e-7;


int n,m,k;

void solve()
{
    cin >> n >> m >> k;
    bool f=1;
    int cnt = n/k,last = n - cnt * k ;
    // 可以分成完整的几段，最后一段如果存在的话剩余多少 
    int x;
    for(int i=1;i<=m;i ++ )
    {
        cin >> x;                
        if(x > cnt + 1) f= 0;//当前数量大于 区间个数的 最大值 
        if(x == cnt + 1)last -- ;//需要用到最后一个不完整的区间 
        if(last < 0 ) f= 0;//最后一个不完整的区间不够用了 
    }
    if(!f) cout << "NO\n";
    else cout <<"YES\n";
}


signed main()
{
//    freopen("1.txt","r",stdin);
    ios
    LL T=1;
    cin>>T;
    while(T -- )
    {
        solve();
    }



    return 0;
}