codeforces 509 B. Painting Pebbles （构造）

题目链接

题意：n堆石子，每堆a[i]个，k种颜色。给每个石子涂色，要求对于每种颜色，任意两堆中该颜色石子的个数最多差一个。问是否有解，有解输出一组方案。

思路：我们发现有解与否只和最大值最小值有关。

设mn为最小值,mx为最大值。

当mx>mn+k时无解，否则有解。

如果有解。。。输出方案就好了。。。

/* ***********************************************
Author :111qqz
Created Time :2016年10月04日 星期二 02时40分20秒
File Name :code/cf/problem/509B.cpp
************************************************ */

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <deque>
#include <map>
#include <string>
#include <cmath>
#include <cstdlib>
#include <bitset>
#define fst first
#define sec second
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define ms(a,x) memset(a,x,sizeof(a))
typedef long long LL;
#define pi pair < int ,int >
#define MP make_pair

using namespace std;
const double eps = 1E-8;
const int dx4[4]={1,0,0,-1};
const int dy4[4]={0,-1,1,0};
const int inf = 0x3f3f3f3f;
int n,k;
int a[105];
int main()
{
	#ifndef  ONLINE_JUDGE 
	freopen("code/in.txt","r",stdin);
  #endif

	cin>>n>>k;
	int mn = 101;
	int mx = -1;
	for ( int i = 1 ; i <= n ; i++)
	{
	    cin>>a[i];
	    mn = min(mn,a[i]);
	    mx = max(mx,a[i]);
	}
	if (mx>mn+k)
	{
	    puts("NO");
	}
	else
	{
	    puts("YES");
	    for ( int i = 1 ; i  <= n ; i++)
	    {
		for ( int j = 1 ; j <= a[i] ; j++) printf("%d ",j%k+1);
		printf("\n");
	    }
	}

  #ifndef ONLINE_JUDGE  
  fclose(stdin);
  #endif
    return 0;
}