背景：开始没有仔细想，后来@liu_jc 说简单遂1Y

思路：找出最大堆的鹅卵石数为max，最小堆数为min。如果max-min>=k,则成立。

          证明：对最大堆编号为：a1,a2,a3~amin-1,amin~amax .对最小堆编号为：b1,b2~bmin.

                    让a1和b1，a2和b2,......,amin和bmin颜色一样。

                    对于剩下的amin+1~amax 鹅卵石不能出现重复颜色，一旦出现就会是2-0>1.

                    所以：剩下的石头数必须小于k.    得证。

          至于其他鹅卵石数在最小和最大数之间的很容易证明可行。

          至于对每一个鹅卵石的涂色，直接循环涂1~k,即可，这样最均衡。

学习：比赛的时候要深入思考，必须要用草稿纸，在头脑里空想是不够的。

我的代码：

#include<stdio.h>
#include<string.h>

int main(void){<span id="transmark"></span>
    int t,k,str[100];
    while(~scanf("%d%d",&t,&k)){
        int min=100000,max=0;
        for(int i=0;i < t;i++){
            scanf("%d",&str[i]);
            if(str[i] < min) min=str[i];
            if(str[i] > max)  max=str[i];
        }
        if(max-min <= k){
            printf("YES\n");
            for(int i=0;i < t;i++){
                for(int j=0,kk=0;j < str[i];j++,kk++){
                    if(j) printf(" ");
                    printf("%d",kk%k+1);
                }
                printf("\n");
            }
        }else printf("NO\n");

    }
	  return 0;
}

Coderforces 509B