[Softeer] 나무 공격

문제링크

---

나무 공격

풀이과정

---

이렇게 푸는게 맞나 .. 싶지만 ?

2차원 배열로 맵을 받은 후에 공격해야할 행의 범위가 입력되면 그 부분만 for문을 통해 돌리자.
if문으로 [j][k] 부분이 1이라면 0으로 바꿔주고, 전체 환경파괴범의 숫자에서 1을 빼주고 for문을 탈출시킨다.

그리고 마지막으로 환경파괴범의 숫자를 출력하면 된다.

코드

---

result = 0

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

pos = [[] for _ in range(n)]

for i in range(n):
    pos[i] = list(map(int, input().split()))
    result += sum(pos[i])

for i in range(2):
    sl, el = map(int, input().split())

    for j in range(sl-1, el):
        for k in range(m):
            if pos[j][k] == 1:
                result -= 1
                pos[j][k] = 0
                break
  
print(result)

결과

---