파이썬 알고리즘 - 패턴문제

파이썬 알고리즘

Python 알고리즘 다른 문제

- 숫자 배열 알고리즘

단어 패턴

문자열(패턴) 하나와 문자열의 배열 하나가 주어집니다.

패턴 문자열의 각각의 문자 하나는, 두번째 문자열 배열의 각각의 문자열 하나에 대응 될 수 있습니다.

해당 배열이 해당 패턴으로 표현 되는지 아닌지의 여부를 확인하는 함수를 만들어 보세요.

예를 들어서, aabb 와 ['abc', 'abc', 'fgh', 'fgh'] 가 주어졌을 경우에는 함수가 True를 반환해야 합니다. 이 경우에는 a가 abc에, b가 fgh에 대응되도록 하면 배열을 해당 패턴으로 표현 하는 것이 가능하기 때문이죠.

반면, aabb 와 ['abc', 'fgh', 'abc', 'fgh'] 가 주어졌을 경우에는 함수가 False를 반환해야 합니다.

모든 문자는 영어 소문자라고 가정합니다.

제출 코드

def wordPattern(pattern, strList):

    patternSet = set(pattern)
    strListSet = set(strList)
    patternSetList = list(patternSet)
    strListSetList = list(strListSet)

    if len(pattern) != len(strList):
        return False

    if len(patternSetList) != len(strListSetList):
        return False

    patterSetListHash = {}

    for i in range(len(patternSetList)):
        patterSetListHash[patternSetList[i]] = strListSetList[i]

    strListArray = []
    for i in range(len(strList)):
        for ch in patterSetListHash:
            if strList[i] == patterSetListHash[ch]:
                strListArray.append(ch)

    strListArrayStr = ""
    for i in strListArray:
        strListArrayStr += i

    strListHash = {}
    patternHash = {}

    rerangeIndex = 0
    for ch in strList:
        if ch not in strListHash:
            strListHash[ch] = rerangeIndex
            rerangeIndex += 1

    rerangeIndex = 0
    for ch in pattern:
        if ch not in patternHash:
            patternHash[ch] = rerangeIndex
            rerangeIndex += 1

    strListCompare = []
    patternCompare = []

    for i in strList:
        for ch in strListHash:
            if i == ch:
                strListCompare.append(strListHash.get(ch))

    for i in pattern:
        for ch in patternHash:
            if i == ch:
                patternCompare.append(patternHash.get(ch))

    if strListCompare == patternCompare:
        return True
    else:
        return False

def main():
    print(wordPattern("aabbccdc"
                      , ["abc", "abc", "ddd", "ddd", "asd", "asd", "d", "asd"]))  # should return True
    print(wordPattern("abab"
                      , ["asdf", "asdf", "fdfd", "fdfd"]))  # should return False

if __name__ == "__main__":
    main()