1. 程式人生 > >實現陣列元素選與不選問題-python實現

實現陣列元素選與不選問題-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裡強制轉換了