[プログラマー/python]第2級英語のドラゴンボール



https://programmers.co.kr/learn/courses/30/lessons/12981

問題を解く


2つの条件を大まかに考慮しなければならない.
1.接話尾規則を確立する.(non->nowはX)
2.出てきた単語は二度と出てこない.(戦車2号)
入出力例からヒントが得られ,誰もがディクシャナで区別された.
単語を次から次へと置き、ルールに合わない単語やすでに出現している単語であれば、その単語の[番号、順序]を返します.
ex)
...
{1: ['tank', 'wheel', 'mother'], 2: ['kick', 'land', 'robot'], 3: ['know', 'dream']}

ソースコード

def solution(n, words):
    people={i:[] for i in range(1,n+1)}
    visited=[words[0]]
    
    people[1].append(words[0])
    for i in range(1,len(words)):
        if words[i-1][-1]==words[i][0] and words[i] not in visited:
            people[i%n+1].append(words[i])
            visited.append(words[i])
            #print(people)
        else:
            return [i%n+1,i//n+1]
    
    return [0,0]