Search code examples
pythonlistrecursionoutputhamming-distance

Appending a List and Recursion (Hamming Distance) - Python 3


I'm supposed to write a program that takes a string of binary code and a number, and outputs all the strings within that hamming distance of the original string. I have a function that does everything, but in the output there are lists within lists. I understand why this is - the function is recursive, and sometimes it will return a list of possible values.

The problem is, I don't know how to change it so it outputs complete strings. For example, for a string of "0000" and hamming distance "2", I get: ["1['100', '010', '001']", "01['10', '01']", "001['1']"]

But what I want is:

1100
1010
1001
0110
0101
0011

Code is below:

def hamming(num, dist):
    if dist == 0:
        return num
    else:
        outputlist = list()
        for item in range(len(num)):
            if len(num[item:]) > dist - 1:
                if num[item] == "0":
                    restoflist = hamming(num[item + 1:], dist - 1)
                    outputlist.append(num[:item] + "1" + str(restoflist))
                else:
                    restoflist = hamming(num[item + 1:], dist - 1)
                    outputlist.append(num[:item] + "0" + str(restoflist))                
        return outputlist

Solution

  • If you loop through the list that is returned by hamming you should be able to append the strings together the right way - like this:

    def hamming(num, dist):
        if dist == 0:
            return num
        else:
            outputlist = list()
            for item in range(len(num)):
                if len(num[item:]) > dist - 1:
                    if num[item] == "0":
                        restoflist = hamming(num[item + 1:], dist - 1)
                        if type(restoflist) is not str
                            for rest in restoflist:
                                outputlist.append(num[:item] + "1" + str(rest))
                        else
                            outputlist.append(num[:item] + "1" + str(restoflist))
                    else:
                        restoflist = hamming(num[item + 1:], dist - 1)
                        if type(restoflist) is not str
                            for rest in restoflist:
                                outputlist.append(num[:item] + "0" + str(rest))
                        else
                            outputlist.append(num[:item] + "0" + str(restoflist))                
        return outputlist