알고리즘/백준(BOJ)

[백준/BOJ]#15805:트리 나라 관광 가이드[트리/파이썬/python]

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

 

15805번: 트리 나라 관광 가이드

윤호는 K개의 도시들이 트리 형태로 연결되어 있는 트리 나라의 관광 가이드이다. 윤호가 새롭게 맡게 된 패키지는 트리 나라의 루트 도시에서 시작해서 모든 도시를 순회하고 오는 상품이다.

www.acmicpc.net

파이썬 코드 

#15805_트리나라관광가이드
from collections import Counter
n=int(input())
city=list(map(int,input().split()))
cntcity=Counter(city)
visited=[0 for _ in range(len(cntcity.keys()))]

parent=[0 for _ in range(len(cntcity.keys()))]
parent[city[0]]=-1

tree=[city[0]]
for i in range(n):

    if visited[city[i]]==0:
        if i<=0:
            parent[city[i]]=-1
        else:
            parent[city[i]]=city[i-1]
        tree.append(city[i])
        visited[city[i]]=1


print(len(cntcity.keys()))
for i in parent:
    print(i, end=' ')