B. Painting Pebbles

题目链接：http://codeforces.com/problemset/problem/509/B

题意：有n堆石头，每堆有ai个,有k种颜色要求给所有石头涂上色。要求任意两堆石头中相同颜色的石头的数量之差小于等于1。 1<=n,k,ai<=100

思路：

这题本质上其实就是[贪心]

1. 考虑到求出最小的一堆石子数量，设为mina

2. 然后将每堆的前mina+1个石子（能染多少是多少）染成1（方便起见）颜色。

3. 后面的石子按顺序染颜色2~k即可。

例如[样例3]：

1. 先求出mina=min{3,2,4,3,5}=2
   

2. 然后按照[步骤2]将每堆的前三颗石子染成11颜色，结果如下：

   1 1 1
   1 1
   1 1 1 ?
   1 1 1
   1 1 1 ? ?

   其中?表示还未染色的石子

3. 然后按照[步骤3]染即可，就成功了（注意此题SPJ）

   1 1 1
   1 1
   1 1 1 2
   1 1 1
   1 1 1 2 3

不可能的情况也很简单，仔细想想就知道当且仅当任意一堆染得石子数量不足本队石子总数（即mina+k+1<a[i]），才输出NO。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;
const int maxn=2e5+7;

int n,k;
int arr[maxn];
int vis[maxn];
int Print[maxn][maxn];

void draw(int num,int id){
    for (int i=0;i<n;i++){
        int cnt = num;
       for (;vis[i]<arr[i] && cnt;vis[i]++,cnt--){
            Print[i][vis[i]] = id;
        }
    }
}

int main(){
    scanf("%d%d",&n,&k);
    int minxx = 0x3f3f3f3f;
    for (int i=0;i<n;i++) {
        scanf("%d", &arr[i]);
        if (arr[i] < minxx) {
            minxx = arr[i];
        }
    }
        draw(minxx+1,1);
        for (int i=2;i<=k;i++){
            draw(1,i);
        }
        for (int i=0;i<n;i++){
            if (vis[i] != arr[i]){
                printf("NO
");
                return 0;
            }
        }
        printf("YES
");
        for (int i=0;i<n;i++){
            for (int j=0;Print[i][j]!=0;j++){
                printf("%d ",Print[i][j]);
            }
            printf("
");
        }
        return 0;
}