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

最新推荐文章于 2023-02-10 14:32:47 发布

原创最新推荐文章于 2023-02-10 14:32:47 发布 · 468 阅读

CC 4.0 BY-SA版权

博客围绕固定数组{0,1,2,3,4,5,6,7,8,9}展开，根据输入的bool数组{0,1,1,1,1,1,1,1,0,0}确定元素是否必须出现，求解所有可能组合，采用回溯法解决组合问题，最后将组合转化为字符串并升序排序，给出了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里强制转换了