파괴되지 않은 건물 (누적합, numpy)

2023. 6. 27. 21:07·Coding Test/programmers
728x90
"""
1 ≤ board의 행의 길이 (= N) ≤ 1,000
1 ≤ board의 열의 길이 (= M) ≤ 1,000
1 ≤ skill의 행의 길이 ≤ 250,000
"""
import numpy as np
def solution(board, skill):
    answer = 0  # 적의 공격 혹은 아군의 회복 스킬이 모두 끝난 뒤 파괴되지 않은 건물의 개수
    n=len(board)
    m=len(board[0])
    temp=[[0 for j in range(m+1)] for i in range(n+1)]

    for t,r1,c1,r2,c2,d in skill:
        tt= -d if t==1 else d
        temp[r1][c1]+=tt
        temp[r1][c2+1]-=tt
        temp[r2+1][c1]-=tt
        temp[r2+1][c2+1]+=tt

    temp = np.cumsum(temp,axis=1)
    temp = np.cumsum(temp,axis=0)
    answer = np.array(board) + temp[:len(board), :len(board[0])]

    return len(np.where(answer > 0)[0])
728x90
저작자표시 비영리 변경금지 (새창열림)
'Coding Test/programmers' 카테고리의 다른 글
  • 순위검색 (해시, 딕셔너리)
  • 1,2,3 떨어뜨리기 (트리, 구현)
  • 파괴되지 않은 건물 (누적합, 2차원 합배열)
  • 주차 요금 계산 (구현)
Karla Ko
Karla Ko
𝘾𝙤𝙣𝙩𝙞𝙣𝙪𝙤𝙪𝙨𝙡𝙮 𝙄𝙢𝙥𝙧𝙤𝙫𝙞𝙣𝙜, 𝘾𝙤𝙣𝙨𝙩𝙖𝙣𝙩𝙡𝙮 𝘿𝙚𝙫𝙚𝙡𝙤𝙥𝙞𝙣𝙜 𝙔𝙚𝙨!
    250x250
  • Karla Ko
    karlaLog
    Karla Ko
  • 전체
    오늘
    어제
    • Total (467)
      • Spring (19)
      • JPA (4)
      • Cloud & Architecture (15)
        • Kubernetes (5)
        • Docker (3)
        • MSA (2)
        • GCP (1)
        • AWS (4)
      • Devops (1)
      • Message Queue (4)
        • Kafka (2)
        • RabbitMQ (2)
      • Git (4)
      • DB (4)
      • Java (9)
      • Python (4)
      • CS (11)
        • OS (8)
        • Network (2)
        • Algorithm (1)
      • Coding Test (392)
        • programmers (156)
        • Graph (43)
        • DP (37)
        • Search (31)
        • Tree (13)
        • Data Structure (26)
        • Combination (12)
        • Implement (18)
        • Geedy (23)
        • Sort (7)
        • Math (21)
        • geometry (2)
  • 블로그 메뉴

    • 홈
  • 링크

  • 공지사항

  • 인기 글

  • 태그

    자료구조
    LIS
    덱
    정렬
    백준
    구현
    큐
    플로이드워셜
    트리
    동적계획법
    프로그래머스
    최대공약수
    알고리즘
    그리디
    구간합
    그래프
    최단거리
    이분탐색
    최소신장트리
    힙
    BFS
    DFS
    재귀
    월간코드챌린지
    조합
    파이썬
    다익스트라
    DP
    스택
    Algorithm
  • hELLO· Designed By정상우.v4.10.3
Karla Ko
파괴되지 않은 건물 (누적합, numpy)
상단으로

티스토리툴바