实现数组元素选与不选问题-python实现

问题描述：

对于固定数组{0,1,2,3,4,5,6,7,8,9}

输入bool数组{0,1,1,1,1,1,1,1,0,0},其中0对应的下标数组元素可出现也可以不出现，1必须出现

出现 所有的可能的组合（组合问题标准的解法是回溯），转化为字符串，并按照字符串升序排序！

#include <stdio.h> #include <iostream> #include <vector> #include <string> #include <algorithm> using namespace std; vector<string> res; void dfs(int* m, int* n, int index,string ss) {     while (index<10&&n[index] == 1)     {         ss += string(1, char(m[index] + '0'));         index++;     }     if (index == 10)     {         res.push_back(ss);         return;     }     dfs(m, n, index + 1, ss);     ss += string(1, char(m[index] + '0'));     dfs(m, n, index + 1, ss); } int main() {     int m[10] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };     int n[10];     for (int i = 0; i < 10; ++i)         cin >> n[i];     dfs(m, n, 0, string());     sort(res.begin(), res.end());     for (int i = 0; i < (int)res.size(); ++i)         cout << res[i] << endl;     return 0; }

搞了半天写了个python实现，用的不熟练

#!/usr/bin/python # -*- coding: utf-8 -*- import sys #a_str = list('0123456789') a_str = '0123456789' print type(a_str) print a_str res =[] bool_list = sys.stdin.readline().split()[0] def dfs(a_str,bool_list,index,tmp):          while index<10 and int(bool_list[index])==1:         tmp += a_str[index]         index +=1     if index ==10:         res.append(tmp)         return     dfs(a_str,bool_list,index+1,tmp)     tmp +=  a_str[index]     dfs(a_str,bool_list,index+1,tmp) dfs(a_str,bool_list,0,'') print 'len:' + str(len(res)) print res #将字符串转换成数字列表 bool_list =list(bool_list) print bool_list bool_list = [int(x) for x in bool_list] print bool_list '''上题也可以先将字符串列表转换成int数组，之后就不用再while里强制转换了