公交车

1009: 公交车
时间限制: 1 Sec 内存限制: 128 MB
题目描述
最近pashmak被运输公司录用了，公司有k辆车，现在有个学校n个学生要用车，学校要将这些学生送到d个不同地方，每个地方呆1天，总共要d天。每天公司要提供车运输这些学生，pashmak就要安排方案。学校为了防止这些学生太过亲密，不希望有任意两个学生d天内都是同一辆车。问pashmak能否合理安排。

注意，我们假设每辆车的容量是无限的。

输入
输入3个数,n, k, d (1 <= n, d <= 1000; 1 <= k <= 10 ^9)

输出
如果可以输出”yes”

然后输出n行，每行d个数，每行输出每个人的安排车辆(车辆编号是从1到k)

方案有多种，这里要求输出字典序最小的一种

不可以输出”no”

样例输入
3 2 2
样例输出
yes
1 1
1 2
2 1
提示
输入

3 2 1

输出

no

来源
cf459C

---

这道题可以简化为k^d与n的关系
如果小于，那么输出no
不然就输出yes
输出方案很简单
先将一个a数组全赋值为1
然后从尾部不断加一
维护k进制即可

#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>#define ll long longusing namespace std;int n,k,d;int a[1005];int main(){    scanf("%d%d%d",&n,&k,&d);    ll kd=1;    for (int i=1;i<=d;i++)    {        kd=kd*k;        if (kd>=n) break;    }    if (kd<n)    {        printf("no\n");        return 0;    }    printf("yes\n");    for (int j=1;j<=d;j++)        a[j]=1;    for (int i=1;i<=n;i++)    {        for (int j=1;j<d;j++)            printf("%d ",a[j]);        printf("%d\n",a[d]);        int x=d;        a[x]++;        while (a[x]>k)        {            a[x--]=1;            a[x]++;        }    }    return 0;}