这题如果将最终的结果竖着看,每一列构成的数可以看成是k进制的数,一共有d列,任意两列都不相同,所以这就是一个d位k进制数全排列的问题,一共有k ^ d个排列,如果k ^ d < n,则打印-1。
打印最终结果时设第一列就为1 1 1 1 ... 1,然后依次每列增加1后(公交车编号从1开始,不是从0开始) ,注意,这里是k进制。
#include <stdlib.h>
#include <stdio.h>
#include <algorithm>
#include <vector>
#include <math.h>
using namespace std;
int tmpvalue[1001][1001];
int main()
{
#ifdef _DEBUG
freopen("d:\\in.txt", "r", stdin);
#endif
int n, k, d;
scanf("%d %d %d", &n, &k, &d);
if (k < n)
{
int total = 1;
for (int i = 1; i <= d;i++)
{
total *= k;
if (total >= n)
{
break;
}
}
if (total < n)
{
printf("-1\n");
return 0;
}
}
for (int i = 1; i <= d;i++)
{
tmpvalue[i][1] = 1;
}
for (int i = 2; i <= n;i++)
{
for (int j = 1; j <= d;j++)
{
tmpvalue[j][i] = tmpvalue[j][i - 1];
}
tmpvalue[1][i] += 1;
for (int j = 1; j <= d; j++)
{
if (tmpvalue[j][i] > k)
{
tmpvalue[j][i] = 1;
tmpvalue[j + 1][i]++;
}
else
break;
}
}
for (int i = 1; i <= d;i++)
{
for (int j = 1; j <= n;j++)
{
printf("%d ", tmpvalue[i][j]);
}
printf("\n");
}
return 0;
}