【水题】Codeforces Round #289 509B Painting Pebbles

点击打开链接

题意：给N堆的每一块石头涂色，令 i 堆中的 k 颜色 表示为 Cik

要求 |Cik-Cjk| <=1

贪心下就可以了

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <string>
#include <iostream>
#include <algorithm>
#include <sstream>
#include <cmath>
using namespace std;
#include <queue>
#include <stack>
#include <set>
#include <vector>
#include <deque>
#include <map>
#define cler(arr, val)    memset(arr, val, sizeof(arr))
typedef long long  LL;
const int MAXN = 120000;
const int MAXM = 140000;
const int INF = 0x3f3f3f3f;
const int mod = 1000000007;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int a[1000],b[122][122],num[122][122];
int main()
{
    int n,k;
    cin>>n>>k;
    cler(b,0);
    for(int i=1;i<=n;i++)
        cin>>a[i];
    for(int i=1;i<=n;i++)
    {
        int tol=0;
        for(int j=1;j<=a[i];j++)
        {
            b[i][j]=tol%k;
            num[i][tol%k]++;
            tol++;
        }
    }
    int flag=0;
    for(int i=1;i<=n;i++)
        for(int j=i+1;j<=n;j++)
            for(int t=0;t<k;t++)
                if(abs(num[i][t]-num[j][t])>1) flag=1;
    if(flag) puts("NO");
    else
    {
        puts("YES");
        for(int i=1;i<=n;i++)
        {
            printf("%d",b[i][1]+1);
            for(int j=2;j<=a[i];j++)
                printf(" %d",b[i][j]+1);
            printf("\n");
        }
    }
    return 0;
}