본문 바로가기
coding 📟/coding test ⌨️

[Softeer] 나무공격 python

by 고돌한 데이터 사이언스 2024. 9. 27.
반응형
import sys
from collections import deque
# 투사체 환경파괴범이 만나면 동시에 사라짐, 만나지 않을 경우 사라짐
# input n,m 초기 격자 칸에는 0, 1 1은 환경파괴범
# 왼쪽 방향에서 공격, 총 2회 공격
# L행 부터 R까지의 구간에 만들어 진행, L과 R의 차이는 항상 4 (항상 5개가 만들어짐))
# 2번의 공격 이후 남아있는 환경 파괴범을 구함

if __name__ == '__main__':

    # input
    m,n = map(int, input().split())
    li = []
    for _ in range(m):
        unit = list(map(int, input().split()))
        li.append(unit)
    first = list(map(int, input().split()))
    second = list(map(int, input().split()))
    att = [first, second]

    # logic
    for i in att:
        for j in range(i[0],i[1]+1):
            temp = li[j-1]

            for idx, k in enumerate(temp):
                if k == 1:
                    temp[idx] = 0
                    break
                    
            # print('''------------------''')
            # print(temp)
            li[j-1] = list(temp)
        
    # 점검
    # print(li)
    cnt = 0
    for i in li:
        try:
            s = sum(i)
            if s != 0:
                cnt += s
            else:
                continue
        except:
            continue

    print(cnt)
반응형

'coding 📟 > coding test ⌨️' 카테고리의 다른 글

[Softeer] 8단 변속기  (0) 2023.08.21
[Softeer] 금고털이  (0) 2023.08.21
[Softeer] A+B  (0) 2023.08.20
[Softeer] 근무 시간  (0) 2023.08.20

댓글