前言

我好菜啊

传送门: https://www.luogu.com.cn/problem/CF509B

思路

要求任意两堆石头中相同颜色的石头的数量之差小于等于1

因此不难想,我们可以贪心地做 :

• 首先把所有石子都涂成相同的颜色 比如: ‘1’

但是问题就来了:
其他颜色还没用到,这么贪心 大概率是错的

• 因此我们只需要把 minn +1 的石子 全涂成相同的颜色
• 然后剩下的 我们就把剩下的颜色涂上去

贪心行不行呢? 交一发再说

CODE

#include <bits/stdc++.h>
using namespace std;
const int N = 100;
int a[N],k,n;
int minn  = 0x3f3f3f3f;
void solve()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        minn = min(a[i],minn);
    }
    for(int i=1;i<=n;i++)
    {
        if(minn+1+k <= a[i])
        {
            cout<<"NO"<<endl;
            return;
        }
    }

    cout<<"YES"<<endl;

    for(int i=1;i<=n;i++)
    {
        int d = 2;
        for(int j=1;j<=a[i];j++)
        {
            if(j<=minn+1)
            {
                cout<<1<<" ";
            }
            else
            {
                cout<<d<<" ";
                ++d;
            }
        }
        cout<<endl;
    }

}
int main()
{
    ios::sync_with_stdio(false);
    solve();
    return 0;
}