42-gjsk132

[gjsk132]

🔗 문제 링크

백준1034 : 램프

✔️ 소요된 시간

20분

✨ 수도 코드

🎯 문제 이해

문제 요약
탁자의 가로 세로가 주어지고, 그 위에 전구가 켜졌는지 아닌지를 주어진다.
스위치를 켜고 끄는데, 주어진 횟수만큼 한 열의 스위치 상태를 반대로 바꾼다.
전구가 모두 켜져있는 행의 최대값을 구하면 된다.

한 번 바꿀 때, 하나의 열의 모든 전구의 상태가 바뀐다.

로직을 알고나면 간단하게 풀 수 있다.

열의 상태가 다르면, 전구를 몇 번 키더라도 둘 다 켜지는 상태가 될 수 없다.
쉽게 말해
[1, 0, 0]과 [0,1,0]이 있을 때,
하나를 [1,1,1]로 만들면 다른 하나는 다 킬 수 없다는 말이다.
따라서, 열의 상태가 같은 것들끼리 묶어서 확인하면 된다.

그 후에 갯수가 많은 것부터 확인하면서, 해당 상태가 횟수를 다 썼을 때, 전원 다 킬 수 있는지만 확인하면 된다.

🔍코드 설명

table = defaultdict(int)

for _ in range(n):
    row = input().rstrip()
    table[row] += 1

각 열 정보를 받고, 열을 key로 하는 dict의 value에 +1 해주어, 상태를 카운트해준다.

table = list(table.items())
table.sort(key=lambda x: -x[1])

dict를 list로 바꾸고, 빈도가 더 높은 순서대로 정렬해준다.

for row, cnt in table:
    need_switch_on = m - sum(list(map(int,row)))
    
    if turn_on_off_cnt < need_switch_on:
        continue
    
    if (turn_on_off_cnt - need_switch_on)%2==1:
        continue
    
    answer = cnt
    break

빈도가 높은 순으로 순회한다.
만약 전구 상태가 전원을 다 키기 위한 횟수가 부족하다면, 다음 전구 상태를 확인한다.
또, 전구 다 키고 남은 횟수가 홀수라면, 전원은 다 킬 수 없기 때문에, 다음 전구 상태를 확인한다.

맨 처음 결과가 나온 값이 가장 큰 값이기 때문에 answer에 값을 넣고 for문은 break해준다.

그 이후 정답을 출력해주면 된다.

from collections import defaultdict

input = open(0).readline

n, m = map(int,input().split())

answer = 0

table = defaultdict(int)

for _ in range(n):
    row = input().rstrip()
    table[row] += 1
    
turn_on_off_cnt = int(input())

table = list(table.items())
table.sort(key=lambda x: -x[1])

for row, cnt in table:
    need_switch_on = m - sum(list(map(int,row)))
    
    if turn_on_off_cnt < need_switch_on:
        continue
    
    if (turn_on_off_cnt - need_switch_on)%2==1:
        continue
    
    answer = cnt
    break

print(answer)

📚 새롭게 알게된 내용

오랜만에 복귀... 🫠