글 읽기 - 5567 결혼식 어디가 틀렸나요..?

qkrdmstlr3 4년 전 0

흠.... 배열 두개 이용해서 풀었는데 어디가 잘못되었는지 모르겠네요... 도와주세요~ㅠ

import sys

n = int(sys.stdin.readline())
li = []
friend, friend_friend = [], []
for i in range(int(sys.stdin.readline())):
    li.append(list(map(int, sys.stdin.readline().split())))
li.sort()  # 순서 안 헷갈리려고 정렬 > 그냥 친구 먼저 넣을려고
for i in range(len(li)):
    if li[i][0] == 1:  # 그냥 친구일때
        friend.append(li[i][1])
    elif li[i][0] in friend:  # 친구의 친구일 때 a가 먼저 나올때
        if li[i][1] in friend:
            continue
        friend_friend.append(li[i][1])
    elif li[i][1] in friend:  # 친구의 친구일 때 b가 먼저 나올때
        if li[i][0] in friend:
            continue
        friend_friend.append(li[i][1])
print(len(friend_friend) + len(friend))

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

5567 결혼식 어디가 틀렸나요..?

qkrdmstlr3 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말