사용자 도구

사이트 도구


ps:problems:boj:5508

Party Lamps

ps
링크acmicpc.net/…
출처BOJ
문제 번호5508
문제명Party Lamps
레벨골드 1
분류

애드혹

시간복잡도O(n)
인풋사이즈n<=100
사용한 언어Python 3.11
제출기록31256KB / 44ms
최고기록40ms
해결날짜2023/07/19

풀이

  • 버튼들이 2개단위 또는 3개 단위의 램프들을 바꾸므로, 이 버튼들의 조합으로 만들어지는 전구의 패턴은 6개 단위로 반복되는 패턴이다.
  • 정확히는 '111111', '000000','010101','101010','011011','100100','110001','001110' 의 8가지 패턴이 가능하다
  • 켜져있는 전구와 꺼져있는 전구의 번호들이 주어졌을때 특정 패턴에 해당하는지는, 번호를 6으로 나눈 나머지를 패턴과 비교해보면 된다.
  • 각 패턴들을 만들기 위해 버튼을 눌러야하는 총 횟수를 생각해보자. '111111'은 1번버튼을 2번 눌러서 총 두번만에, 또는 1,2,3번 버튼을 한번씩 눌러서 3번만에, 아니면 이후에 같은 버튼을 2n번 더 눌러서 만들수 있다. 불가능한 경우는 버튼을 누른 총 횟수가 1번뿐일 때이다. '000000'은 어떤 횟수로든 다 만들수 있다. 이런식으로 패턴마다 따져보면, '100100','110001','001110' 은 두번 이상 버튼을 눌러야 만들수 있고, '011011'은 한번만 누르든가 아니면 세번 이상 눌러야 한다.
  • 이제 끝. 버튼누른 횟수, 켜진 전구번호, 꺼진 전구번호를 갖고서 8개의 패턴들 각각에 대해서 만들수 있는 패턴인지 아닌지를 체크하면 된다.
  • 전구 번호들은 O(1)개, 패턴 갯수도 O(1)개라서 계산은 O(1)에 다 끝난다. 다만 최종 출력 결과로 O(n)길이의 문자열을 만들어줘야 하기 때문에 엄밀한 시간복잡도는 O(n)..

코드

"""Solution code for "BOJ 5508. Party Lamps".

- Problem link: https://www.acmicpc.net/problem/5508
- Solution link: http://www.teferi.net/ps/problems/boj/5508

Tags: [ad hoc]
"""

ON = '1'
OFF = '0'
PATTERNS = (
    ('111111', (1,)),
    ('000000', (0,)),
    ('010101', (0,)),
    ('101010', (0,)),
    ('011011', (0, 2)),
    ('100100', (0, 1)),
    ('110001', (0, 1)),
    ('001110', (0, 1)),
)


def main():
    N = int(input())
    C = int(input())
    on_lamps = [int(x) - 1 for x in input().split()][:-1]
    off_lamps = [int(x) - 1 for x in input().split()][:-1]

    for pat, invalid_nums in PATTERNS:
        if (
            C not in invalid_nums
            and all(pat[x % 6] == ON for x in on_lamps)
            and all(pat[x % 6] == OFF for x in off_lamps)
        ):
            print(pat * (N // 6) + pat[: N % 6])


if __name__ == '__main__':
    main()

토론

댓글을 입력하세요:
K D U G S
 
ps/problems/boj/5508.txt · 마지막으로 수정됨: 2023/07/20 05:28 저자 teferi