背景:开始没有仔细想,后来@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;
}