题意:给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;
}