字符串组合算法

时间:2023-01-07 18:43:07

全组合

例如给定字符串“abc”,全组合意思从中去0个元素,1个元素,一直到n个元素,介绍二进制做法。以字符串“abc”为例:
  • 000 <---> NULL
  • 001 <---> c
  • 010 <---> b
  • 011 <---> bc
  • 100 <---> a
  • 101 <---> ac
  • 110 <---> ab
  • 111 <---> abc

思路出来了,代码也比较好写,分享一下我的代码:

/**
* Write a method that returns all subsets of a set
*/

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

/**
* 通过0到2^-1来标识子集
*
* T = (n * 2^n)
*
*/
void getSubset(char *str, int len)
{
int i, max, index, j;

max = 1 << len;

for (i = 1; i < max; i ++) {
j = i;
index = 0;

while (j) {
if (j & 1) {
printf("%c", str[index]);
}
j >>= 1;
index ++;
}
printf("\n");
}
}

int main(void)
{
char str[1000];

while (scanf("%s", str) != EOF) {
getSubset(str, strlen(str));

}

return 0;
}


从n中选m个数


这里分为两种方法:递归和回溯

递归

递归思路如下,从n个数中取出m个数,可以分解为以下两步:
  1. 从n个数中选取编号最大的数,然后在剩下的n-1个数中选取m-1个数。直到从n-(m-1)中选取一个数为止
  2. 从n个数中选取次小的数,重复1的操作

代码如下:

/**
* 递归法解决组合问题
*/
void combine(int *arr, int n, int m, int *tmp, const int M)
{
int i, j;

for (i = n; i >= m; i --) {
tmp[m] = i;
if (m == 0) { // 选出m个数
for (j = 0; j < M; j ++) {
printf("%d ", arr[tmp[j]]);
}
printf("\n");
} else {
combine(arr, i - 1, m - 1, tmp, M);
}
}
}

DFS

其实考虑到用dfs,这道题目就简单很多,dfs的回溯条件就是临时数组的大小==k即可,同时附加一道LeetCode上的题目,用dfs思路ac

题目

Given two integers n and k, return all possible combinations of k numbers out of 1 ... n.

For example,
If n = 4 and k = 2, a solution is:

字符串组合算法

ac代码

public class Solution {
public static ArrayList<ArrayList<Integer>> combine(int n, int k) {
ArrayList<ArrayList<Integer>> rs = new ArrayList<ArrayList<Integer>>();
ArrayList<Integer> list = new ArrayList<Integer>();

dfs(1, k, n, list, rs);

return rs;
}

public static void dfs(int pos, int k, int n, ArrayList<Integer> list, ArrayList<ArrayList<Integer>> rs) {
if (list.size() == k) {
rs.add(new ArrayList<Integer>(list));
}

for (int i = pos; i <= n; i ++) {
list.add(i);
dfs(i + 1, k, n, list, rs);
list.remove(list.size() - 1);
}
}
}