洛谷 B3621 枚举元组

题目描述

n 元组是指由 n 个元素组成的序列。例如(1,1,2) 是一个三元组、(233,254,277,123) 是一个四元组。

给定 n 和 k，请按字典序输出全体 n 元组，其中元组内的元素是在 [1,k] 之间的整数。

「字典序」是指：优先按照第一个元素从小到大的顺序，若第一个元素相同，则按第二个元素从小到大……依此类推。详情参考样例数据。

输入格式

仅一行，两个正整数 n,k。

输出格式

若干行，每行表示一个元组。元组内的元素用空格隔开。

输入输出样例

输入 #1

2 3

输出 #1

1 1
1 2
1 3
2 1
2 2
2 3
3 1
3 2
3 3

输入 #2

3 3

输出 #2

1 1 1
1 1 2
1 1 3
1 2 1
1 2 2
1 2 3
1 3 1
1 3 2
1 3 3
2 1 1
2 1 2
2 1 3
2 2 1
2 2 2
2 2 3
2 3 1
2 3 2
2 3 3
3 1 1
3 1 2
3 1 3
3 2 1
3 2 2
3 2 3
3 3 1
3 3 2
3 3 3

说明/提示

对于 100%100% 的数据，有n≤5,k≤4。

源代码

#include <bits/stdc++.h>
using namespace std;
int n,k;
int a[1000];
void dfs(int num)
{
    if(num==n+1)
    {
        for(int i=1;i<=n;i++)
        {
            cout<<a[i]<<" ";
        }
        cout<<endl;
        return;
    }
    for(int i=1;i<=k;i++)
    {
        a[num]=i;
        dfs(num+1);
    }
}

int main()
{
    cin>>n>>k;
    dfs(1);
    return 0;
}

 

测试点信息

总结 

 题目类似于用1~k来组数，而且数字可以重复使用，用深度优先搜索（DFS）的方法来做：

如果 ：到达边界，就输出这一组数并返回

否则 ：

for(k种情况)

        记录当前i的值

        dfs(下一层）