二维数组排列组合并求和,得到的结果和给定值相等,输出满足求和的所有元素下标。C语言
时间: 2023-06-13 14:02:37 浏览: 128
以下是一个基于回溯法的解法,可以求出所有满足要求的元素下标组合:
```c
#include <stdio.h>
#define ROW 3
#define COL 3
#define TARGET_SUM 10
void find_combination(int arr[][COL], int row, int col, int target_sum, int cur_sum, int cur_row, int cur_col, int visited[][COL], int* count, int* result[][COL]) {
if (cur_sum == target_sum) {
// 找到了一组组合,将其保存到结果集中
int i, j;
for (i = 0; i < row; i++) {
for (j = 0; j < col; j++) {
if (visited[i][j]) {
result[*count][i][j] = 1;
}
}
}
(*count)++;
return;
}
if (cur_row >= row) {
// 行数超出范围,返回
return;
}
if (cur_col >= col) {
// 列数超出范围,转到下一行继续查找
find_combination(arr, row, col, target_sum, cur_sum, cur_row+1, 0, visited, count, result);
return;
}
if (cur_sum + arr[cur_row][cur_col] <= target_sum) {
// 尝试选择当前元素
visited[cur_row][cur_col] = 1;
find_combination(arr, row, col, target_sum, cur_sum+arr[cur_row][cur_col], cur_row, cur_col+1, visited, count, result);
visited[cur_row][cur_col] = 0;
}
// 尝试不选择当前元素
find_combination(arr, row, col, target_sum, cur_sum, cur_row, cur_col+1, visited, count, result);
}
int main() {
int arr[ROW][COL] = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}};
int visited[ROW][COL] = {{0}};
int result[ROW*COL][ROW][COL];
int count = 0;
find_combination(arr, ROW, COL, TARGET_SUM, 0, 0, 0, visited, &count, result);
int i, j, k;
for (i = 0; i < count; i++) {
printf("Combination %d:\n", i+1);
for (j = 0; j < ROW; j++) {
for (k = 0; k < COL; k++) {
if (result[i][j][k]) {
printf("(%d, %d) ", j, k);
}
}
}
printf("\n");
}
return 0;
}
```
在上面的代码中,`find_combination`函数是用来查找元素下标组合的,其参数说明如下:
- `arr`:原始的二维数组;
- `row`:数组的行数;
- `col`:数组的列数;
- `target_sum`:需要求得的目标和;
- `cur_sum`:当前已经求得的和;
- `cur_row`:当前正在处理的行数;
- `cur_col`:当前正在处理的列数;
- `visited`:一个与原始数组大小相同的二维数组,用于记录已经访问过的元素;
- `count`:一个指向整数的指针,用于记录找到的组合数量;
- `result`:一个三维数组,用于保存找到的所有组合的下标信息。
在函数内部,我们首先判断当前是否已经求得目标和,如果是,则将当前的组合信息保存到 `result` 数组中,并将 `count` 值加 1;否则,我们尝试选择当前元素,并递归地继续查找下一个元素;然后,我们再尝试不选择当前元素,并递归地查找下一个元素。
在 `main` 函数中,我们定义了一个原始数组 `arr`,一个记录访问情况的数组 `visited`,一个保存结果的三维数组 `result`,以及一个记录找到组合数量的变量 `count`。然后,我们调用 `find_combination` 函数来查找所有满足要求的组合,最后按顺序输出所有的组合信息即可。
注意:这个解法的时间复杂度是指数级的,对于大型数组的情况,可能会超时或者占用过多的内存,需要根据实际情况进行优化。
阅读全文