【题目链接】:http://codeforces.com/contest/509/problem/B

【题意】



给n鹅卵石染色;

有k种颜色可供选择;

问你有没有染色方案;

使得各个堆的鹅卵石里面,第i颜色的鹅卵石的数目的差的绝对值都不大于1

【题解】



把所有的堆

一开始按照石头的数目升序排一遍;

然后顺序处理

对于第一堆石头

1..k,1..k一直重复直到满a[1]个石头;

这以后

每一行直接复制前一行的;

然后开始给多出来的石头染色;

就给他们染成能够染的石子就好;

一开始1..k都还能再染一个;

所以如果∑a[i]-a[i-1]>k的话就无解

否则肯定有解的;

每次比上一层多a[i]-a[i-1]个石子就好(选择可以染的颜色);

模拟



【Number Of WA】



0



【完整代码】

#include <bits/stdc++.h>

using namespace std;

#define lson l,m,rt<<1

#define rson m+1,r,rt<<1|1

#define LL long long

#define rep1(i,a,b) for (int i = a;i <= b;i++)

#define rep2(i,a,b) for (int i = a;i >= b;i--)

#define mp make_pair

#define ps push_back

#define fi first

#define se second

#define rei(x) cin >> x

#define pri(x) cout << x

#define ms(x,y) memset(x,y,sizeof x)

typedef pair<int,int> pii;

typedef pair<LL,LL> pll;

const int dx[9] = {0,1,-1,0,0,-1,-1,1,1};

const int dy[9] = {0,0,0,-1,1,-1,1,-1,1};

const double pi = acos(-1.0);

const int N = 110;

struct abc

{

    int id,val;

};

int n,k,ma[N],ans[N][N],rest[N];

abc a[N];

bool cmp1(abc a,abc b)

{

    return a.val<b.val;

}

int main()

{

    //freopen("F:\\rush.txt","r",stdin);

    ios::sync_with_stdio(false);

    rei(n),rei(k);

    rep1(i,1,n)

        rei(a[i].val),a[i].id = i;

    rep1(i,1,k)

        rest[i] = 1;

    sort(a+1,a+1+n,cmp1);

    int now = 0;

    rep1(i,1,a[1].val)

    {

        now++;

        if (now>k) now = 1;

        ans[a[1].id][i] = now;

    }

    rep1(i,2,n)

    {

        rep1(j,1,a[i-1].val)

            ans[a[i].id][j] = ans[a[i-1].id][j];

        int need = a[i].val-a[i-1].val;

        int now = 1;

        if (need)

            rep1(j,1,k)

                if (rest[j])

                {

                    rest[j]=0;

                    ans[a[i].id][a[i-1].val+now] = j;

                    need--;now++;

                    if (!need) break;

                }

        if (need) return pri("NO"<<endl),0;

    }

    pri("YES"<<endl);

    rep1(i,1,n)

    {

        for (int j = 1;;j++)

        {

            if (ans[i][j]==0) break;

            pri(ans[i][j]<<' ');

        }

        pri(endl);

    }

    //printf("\n%.2lf sec \n", (double)clock() / CLOCKS_PER_SEC);

    return 0;

}