[C++知识库]B. Painting Pebbles1

// Problem: B. Painting Pebbles
// Contest: Codeforces - Codeforces Round #289 (Div. 2, ACM ICPC Rules)
// URL: https://codeforces.com/problemset/problem/509/B
// Memory Limit: 256 MB
// Time Limit: 1000 ms
// 2022-03-20 17:09:29
// 
// Powered by CP Editor (https://cpeditor.org)

#include<bits/stdc++.h>
using namespace std;

#define rep(i,l,r) for(int i=(l);i<=(r);i++)
#define per(i,l,r) for(int i=(l);i>=(r);i--)
#define ll long long
#define mset(s,t) memset(s,t,sizeof(t))
#define mcpy(s,t) memcpy(s,t,sizeof(t))
#define fi first
#define se second
#define pb push_back
#define all(x) (x).begin(),(x).end()
#define SZ(x) ((int)(x).size())
#define mp make_pair

typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef vector<int> vi;   
typedef vector<ll> Vll;               
typedef vector<pair<int, int> > vpii;
typedef vector<pair<ll, ll> > vpll;                        

const ll mod = 1e9 + 7;
//const ll mod = 998244353;
const double pi  = acos(-1.0);

inline ll qmi (ll a, ll b) {
	ll ans = 1;
	while (b) {
		if (b & 1) ans = ans * a%mod;
		a = a * a %mod;
		b >>= 1;
	}
	return ans;
}
inline int read () {
	int x = 0, f = 0;
	char ch = getchar();
	while (!isdigit(ch)) f |= (ch=='-'),ch= getchar();
	while (isdigit(ch)) x = x * 10 + ch - '0', ch = getchar();
	return f?-x:x;
}
template<typename T> void print(T x) {
	if (x < 0) putchar('-'), x = -x;
	if (x >= 10) print(x/10);
	putchar(x % 10 + '0');
}
inline ll sub (ll a, ll b) {
	return ((a - b ) %mod + mod) %mod;
}
inline ll add (ll a, ll b) {
	return (a + b) %mod;
}
inline ll inv (ll a) {
	return qmi(a, mod - 2);
}
vector<int> a, col[110];
void solve() {
	int n, k;
	cin >> n >> k;
	
	int b[110][110] = {0};
	for (int i = 0; i < n; i ++) {
		int x;
		cin >> x;
		a.pb(x);
		int c = 1;
		
		for (int j = 0; j < a[i]; j ++) {
			col[i].pb(c);
			c ++;
			if (c > k) c = 1;
		}
		for (int j = 0; j < col[i].size(); j ++) {
			b[i][col[i][j]] ++;
		}
	}
	for (int i =  0; i < n; i ++) {
		for (int j = 0; j < n; j ++) {
			for (int m = 1; m <= k; m ++) {
				if (abs(b[i][m] - b[j][m]) > 1) {
					puts("NO");
					return;
				}
			}
		}
	}
	puts("YES");
	for (int i = 0; i < n; i ++) {
		for (auto t: col[i]) {
			cout << t<< ' ';
		}
		puts("");
	}
	
}
int main () {
	// ios::sync_with_stdio(0),cin.tie(0), cout.tie(0);
    int t;
    t =1;
    //cin >> t;
    while (t --) solve();
    return 0;
}

题意:给定n，k，(i, j) b[i][k] - b[j][k] <= 1

题解:按照1 2 3 4 周期性构造就行了