1
2
3
4
5
6
7
8
9
10
11
12
13
| def solution(n, words):
answer = []
dic={}
for i in range(len(words)-1):
if words[i][-1]!=words[i+1][0]:
return [((i+1)%n)+1,((i+1)//n)+1]
elif words[i] in dic:
return [(i%n)+1,(i//n)+1]
else:
dic[words[i]]=1
if words[-1] in dic:
return [((len(words)-1)%n)+1,((len(words)-1)//n)+1]
return [0,0]
|