[Softeer] 나무 공격

✅ 문제

https://softeer.ai/practice/9657

 

문제

환경 파괴범 때문에 화가 난 숲의 요정은 나무 공격을 진행하려 합니다. 나무 공격 진행시 투사체 5개가 생성되어 지정된 방향으로 전진합니다. 각 투사체와 최초로 접촉한 환경 파괴범은 사라지게 되며 이때 투사체 역시 동시에 사라지게 됩니다. 만약 투사체가 환경 파괴범과 마주치지 않는다면 조용히 사라지게 됩니다.

 

이는 n×m 크기의 격자에서 진행됩니다. 초기에 격자의 각 칸에는 숫자 0 또는 1이 적혀있으며 0은 비어있음을, 1은 환경 파괴범이 해당 위치에 서있음을 뜻합니다.

 

 

숲의 요정은 항상 격자의 왼쪽 방향에서 나무 공격을 진행하며, 총 2회 진행합니다. 공격은 특정 행 L부터 행 R까지의 구간에 한하여 투사체를 만들어 진행하게 되며, 모든 투사체는 행 변화 없이 정확히 오른쪽 방향으로만 진행하게 됩니다. 투사체는 처음으로 만나는 환경 파괴범과 함께 사라지게 되며, 끝까지 만나지 않는 투사체는 조용히 사라지게 됩니다. L과 R의 차이는 항상 4이기 때문에 투사체는 항상 5개가 만들어짐에 유의합니다. 다음은 1~5 행에 걸쳐 공격을 진행한 경우입니다.

 

그 다음 두 번째 공격이 2~6 행에 걸쳐 진행되면, 다음과 같은 결과를 얻게 됩니다.

 

 

따라서 위의 예시에서는 2번의 공격 이후 남아있는 서로 다른 환경 파괴범의 수가 2명이 됩니다.

 

초기 격자의 정보와 2번의 공격에 대한 정보가 주어졌을 때, 공격을 순서대로 진행한 이후 격자에 남아있는 서로 다른 환경 파괴범의 수를 구하는 프로그램을 작성해보세요.

제약조건

[조건 1] 5 ≤ n, m ≤ 100

[조건 2] 1 ≤ L1, L2, R1, R2 ≤ n

 

입력

첫 번째 줄에는 격자의 크기를 나타내는 n과 m이 공백을 사이에 두고 주어집니다.

 

두 번째 줄부터는 n개의 줄에 걸쳐 각 행에 해당하는 m개의 격자 정보가 공백을 사이에 두고 주어집니다. 격자 내 숫자는 0, 1로만 주어지며 0은 비어있는 칸을, 1은 환경 파괴범이 있는 칸임을 뜻합니다.

n+2번째 줄에는 첫 번째 공격 정보에 해당하는 L1​,R1​값이 공백을 사이에 두고 주어집니다. 이는 L1​번째 행부터 R1​번째 행까지 공격이 이루어짐을 뜻하며, R1​−L1​은 항상 4임을 가정해도 좋습니다.

n+3번째 줄에는 두 번째 공격 정보에 해당하는 L2​,R2​값이 공백을 사이에 두고 주어집니다. 이는 L2​번째 행부터 R2​번째 행까지 공격이 이루어짐을 뜻하며, R2​−L2​는 항상 4임을 가정해도 좋습니다.

 

출력

첫 번째 줄에 두 번의 공격 이후 격자에 남아있는 서로 다른 환경 파괴범의 수를 출력합니다.

 

✅ 제출 답안

import sys
input = sys.stdin.readline

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

## 환경 파괴범 수를 기록
bad = [0] * n # 이렇게 하면 만약 n이 3이면 bad = [0, 0, 0]
for i in range(n):
    row = list(map(int, input().split()))
    bad[i] = sum(1 for v in row if v == 1) # row에서 1의 개수를 세는 역할로, 1을 발견할 때마다 1을 더해서 총 개수 구함


## 공격!!!!! 두 번의 공격을 수행하여 특정 범위의 행에서 환경 파괴범 수를 줄이기
for _ in range(2): # 공격 두 번 수행

    # 공격 범위 입력 : l번째 행부터 r번째 행까지 파괴범 수 1씩 줄임
    l, r = map(int, input().split())
    
    # 공격 행마다 1씩 빼주고 0이라면 제외
    for i in range(l - 1, r): # l - 1은 인덱스가 0부터 시작하기 때문, range()는 끝 값 r을 포함하지 않으므로 r은 유지지
        if bad[i] > 0: # 파괴범이 존재하는 경우에만 감소 + 파괴범이 음수가 되면 안됨
            bad[i] -= 1 # 파괴범 1명 감소시키기

answer = sum(bad)
print(answer)

 

 

✅ 해설 및 풀이과정

가끔 뜬금없이 map 함수 쓸지 말지에 대해 헷갈린다. 왤까.

numbers = ["1", "2", "3", "4"]
result = list(map(int, numbers))
print(result)  # 출력: [1, 2, 3, 4]

numbers = ["1", "2", "3", "4"]
result = []
for x in numbers:
	result.append(int(x))
print(result)  # 출력: [1, 2, 3, 4]