새소식

PS/LeetCode

3254. Find the Power of K-Size Subarrays I

  • -

Problem : https://leetcode.com/problems/find-the-power-of-k-size-subarrays-i

Difficulty : Medium

Status : Solved

Time : ??:??:??



풀이


어떻게 문제를 분류해 볼 수 있을까.. 싶은 문제.

연속 숫자로 sort된 길이를 임시로 저장해가며, 그 길이가 k 이상이면 그 인덱스의 숫자를 (maximum 값이 현재 index임이 충족된다), 그렇지 않다면 -1을 담아 반환하기만 하면 된다.

기본적인 sort 개념을 어떻게 응용해 볼 수 있을까를 물어보는 문제라고 생각한다.


풀이 코드

class Solution:
    def resultsArray(self, nums: List[int], k: int) -> List[int]:
        sortLen = [1]*len(nums)
        
        for i in range(len(nums)-1) :
            if nums[i]+1 == nums[i+1] :
                sortLen[i+1] = sortLen[i]+1

        return [ nums[i] if sortLen[i] >= k else -1 for i in range(k-1, len(nums)) ]
        
Contents

포스팅 주소를 복사했습니다

이 글이 도움이 되었다면 공감 부탁드립니다.