B3621 枚举元组

坐牢链接

题目描述

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

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

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

输入格式

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

输出格式

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

说明/提示

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

具体思路：

利用递归完成，字典序由for循环从小到大解决，排列数字由dfs递归解决，当生成一组时直接输出并返回探索其他可能。

代码：

1.利用两次for循环，外层从1开始按序赋值，if判断是否满足n元组，若满足则用内层循环输出，否则dfs向下搜索。

#include <bits/stdc++.h>
using namespace std;
int n, k;

void dfs(int a[], int t) {
	for (int i = 1; i <= k; i++) {
		a[t] = i;
		if (t == n - 1) {
			for (int j = 0; j < n; j++) {
				cout << a[j] << ' ';
			}
			cout << endl;
		} else dfs(a, t + 1);
	}
}

int main() {
	cin >> n >> k;
	int a[n];
	memset(a,0,sizeof(a));
	dfs(a, 0);
	return 0;
}

2.更易于理解的方法，判断是否满足，满足则输出并return回上一级，不满足则赋值并且向下深搜。dfs(1)进入递归后首先进行下侧的a[1]=1,a[2]=1,a[3]=1。输出后return并利用for赋值a[3]=2,以此类推。

#include <bits/stdc++.h>
using namespace std;
int a[5];
int n, k;

void dfs(int t) {
	if (t == n + 1) {
		for (int i = 1; i <= n; i++)
			cout << a[i] << " ";
		cout << endl;
		return;
	}
	for (int i = 1; i <= k; i++) {
		a[t] = i;
		dfs(t + 1);
	}
}

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