Algorithm

[프로그래머스] 수열과 구간 쿼리 4 - 파이썬

potato_pizza 2024. 3. 14. 12:18
728x90

Programmers

수열과 구간 쿼리 4

https://school.programmers.co.kr/learn/courses/30/lessons/181922

[프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr](https://school.programmers.co.kr/learn/courses/30/lessons/181922)

문제

코드

<시간 초과 풀이>

def solution(arr, queries):
    answer = []
    for i in range(len(queries)):
        s, e, k = queries[i]
        while s <= i and i <= e:
            if i % k == 0:
                arr[i] += 1
            else: continue
    return answer 

<정답 풀이>

def solution(arr, queries):
    answer = []
    for s, e, k in queries:
        for i in range(s, e+1):
            if i % k == 0:
                arr[i] += 1
    answer = arr
    return answer

Reference

728x90
반응형