codeforces 459C Pashmak and Buses（模拟,组合数A）

#include<iostream>
#include<algorithm>
#include<string>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
using namespace std;
#define ll __int64
int mp[1005][1005];

int main(){
    int n,k,d;
    scanf("%d%d%d",&n,&k,&d);
    
    int ddd=d,kkk=1;

    int flag=0;
    while(ddd--)
    {
        kkk=kkk*k;
        if(kkk>=n)flag=1;
    }

    if(flag==0)
        printf("-1\n");
    else {

        int kk,num=1,kkk=1;
        for(int i=0;i<d;i++)
        {
            kk=1;
            for(int j=0;j<n;j++)
            {
                mp[i][j]=kk;
                if(num==kkk)kk++,kkk=0;
                kkk++;
                if(kk>k)kk=kk-k;
            }
            num=num*k;
        }

        for(int i=0;i<d;i++)
        {
            int yi=0;
            for(int j=0;j<n;j++)
            {
                if(yi)printf(" ");yi=1;
                printf("%d",mp[i][j]);
            }
            puts("");
        }
    }
    return 0;
}