[백준_1411] 비슷한단어_python

https://www.acmicpc.net/problem/1411

1411번: 비슷한 단어

 

문제

만약 어떤 단어A를 숌스럽게 바꿔서 또다른 단어 B로 만든다면, 그 단어는 비슷한 단어라고 한다.

어떤 단어를 숌스럽게 바꾼다는 말은 단어 A에 등장하는 모든 알파벳을 다른 알파벳으로 바꾼다는 소리다. 그리고, 단어에 등장하는 알파벳의 순서는 바뀌지 않는다. 두 개의 다른 알파벳을 하나의 알파벳으로 바꿀 수 없지만, 자기 자신으로 바꾸는 것은 가능하다.

예를 들어, 단어 abca와 zbxz는 비슷하다. 그 이유는 a를 z로 바꾸고, b는 그대로 b, c를 x로 바꾸면, abca가 zbxz가된다.

단어가 여러 개 주어졌을 때, 몇 개의 쌍이 비슷한지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 단어의 개수 N이 주어진다. 둘째 줄부터 N개의 줄에 한 줄에 하나씩 단어가 주어진다. 단어의 길이는 최대 50이고, N은 100보다 작거나 같은 자연수이다. 모든 단어의 길이는 같고, 중복되지 않는다. 또, 알파벳 소문자로만 이루어져 있다.

출력

첫째 줄에 총 몇 개의 쌍이 비슷한지 출력한다.

문제해결

구현 문제이다. 단순구현문제라곤 하지만 아직은 구현이 쉽지않다.

문제해결 아이디어는 다음과 같다. 문제 조건에 따르자면 abca 와 zkdz  는 서로 '비슷한 쌍'이다.

예제를 예로들면 aa, ab, bb, cc, cd가 주어지면 비슷한 쌍의 종류의 묶음은 (aa,bb,cc) 와 (ab,cd)로 나뉠수 있고

(aa,bb), (aa,cc), (bb,cc), (ab,cd) 으로 총 4쌍으로 나눌수 있다. 쌍이라 함은 2가지가 1묶음인 경우를 쌍으로 여긴다.

여기에서 (aa,bb,cc) 와 (ab,cd)는 3C2와 2C2로 조합으로 경우의수를 구할 수 있고 3+1로 총 4쌍이 나온다고 볼 수 있다.

그래서 입력에 주어진 aa,bb,cc가 같은 쌍으로 여겨지고 분리되어 묶여질수 있게하기위해 같은 원소가 나오는건 1로치환이 되고

 

그리고 같은수가 여러개가 주어지는 abba과 czzc 같은 경우엔 1221과 같은 형태로 치환이 되도록 설정을 해주었다.

그리고 같은 종류의 쌍들을 카운트해주기 위해 이차원리스트를 튜플로 바꾸었다. 그리고 set함수를 사용해 중복되지않는 키값을 뽑은 후 그 키값에 해당하는 갯수들을 세어주고나서 조합을 사용해 경우의 수를 합해 정답을 출력해주었다.

 

한고비 넘기면 다른 고비가 계속해 나타났던 문제였다. 구현이 쉽지 않아 계속해 추가하는 방식으로 문제를 해결하다보니 코드가 깔끔하지 않은것 같기도하다.

내가 생각했을 때 이 문제의 키 포인트는 map 을 이용해 리스트를 tuple로 바꿔주는 점에서 많이 찾아보았던것 같다. mutable한 리스트와 immutable한 튜플의 차이점에서 애를 많이 먹었고 알고리즘, 문제푸는 법 뿐만 아니라 언어의 기능적인 면과 구조적으로 더 깊게 공부해야겠다는 생각을 하게되었다.

from itertools import combinations
n = int(input())
arr=[]
res=[]
for i in range(n):
    alpha = list(input())
    t = 0
    for j in range(len(alpha)):
        x = alpha[j]
        if x in alpha:
            for k in range(len(alpha)):
                if x == alpha[k]:
                    alpha[k] = t
            t+=1
    res.append(alpha)
answer=0
a = set(map(tuple,res))
for i in a:
    num=res.count(list(i))
    if num>1:
        ar =[]
        for i in range(num):
            ar.append(i)
        c=len(list(combinations(ar,2)))
        answer += c
print(answer)