[백준] 4354번 문자열 제곱 (파이썬)

4354번: 문자열 제곱

 

 

코드

while 1:
    s=input()
    if s==".":
        break
    else:
        s_len=len(s)
        p_table=[0 for _ in range(s_len)]
 
        j=0
        for i in range(1,s_len):
            while j>0 and s[i]!=s[j]:
                j=p_table[j-1]
            if s[i]==s[j]:
                j+=1
                p_table[i]=j
 
 
        p_len=s_len-p_table[s_len-1]
        
        if s_len % p_len == 0:
            print(s_len//p_len)
        else:
            print(1)