일부 정점의 차수 목록이 있다고 가정합니다. 그래프를 형성하는지 트리를 형성하는지 확인해야 합니다.
따라서 입력이 deg =[2,2,3,1,1,1]과 같으면 출력은 Tree
가 됩니다.
이 문제를 해결하기 위해 다음 단계를 따릅니다. −
- vert :=정점의 수
- deg_sum :=모든 정점의 모든 차수 값의 합
- 2*(vert-1)가 deg_sum과 같으면
- '트리' 반환
- '그래프' 반환
이해를 돕기 위해 다음 구현을 살펴보겠습니다. −
예시 코드
def solve(deg): vert = len(deg) deg_sum = sum(deg) if 2*(vert-1) == deg_sum: return 'Tree' return 'Graph' deg = [2,2,3,1,1,1] print(solve(deg))
입력
[2,2,3,1,1,1]
출력
Tree