[리트코드] 34. Find First and Last Position of Element in Sorted Array

안녕하세요. 개발조각입니다.😊

이렇게 연속으로 두번쓸 생각은 없었는데 어제 안한 것까지 생각해서 두 개를 올려보았습니다.

35번 문제가 easy라서 먼저 풀었는데 왠지 34번도 비슷할 것 같아서 문제만 봐야지 했다가

오? 이건 쉬운 문제다 라는 생각이 들어서 풀어보았습니다.

---

https://leetcode.com/problems/find-first-and-last-position-of-element-in-sorted-array/

Find First and Last Position of Element in Sorted Array - LeetCode

이 문제는 target이 nums배열에 시작점과 끝 값을 구하는 문제입니다.

 

Input: nums = [5,7,7,8,8,10], target = 8
Output: [3,4]

Input: nums = [5,7,7,7,8,10], target = 7
Output: [1,3]

nums배열에 target이 여러개일 경우 시작 값과 끝 값을 찾아주면 됩니다.

 

Input: nums = [5,7,7,8,8,10], target = 5
Output: [0,0]

nums배열에 target이 1개만 존재하는 경우 시작값과 끝 값이 동일하게 됩니다.

 

Input: nums = [5,7,7,8,8,10], target = 6
Output: [-1,-1]

nums배열에 target이 존재하지 않을 경우 [-1, -1]를 넣어주면 됩니다.

---

소스코드

var searchRange = function(nums, target) {
    let idx = nums.findIndex(x=> x === target);
    if(idx === -1) return [-1, -1];
    
    let end = idx;
    while(1){
        if(nums[end] !== nums[end+1]) break;
        end++;
    }
    return [idx, end];
};

먼저 findInesx를 써주어 nums배열에서 target의 인덱스 시작 값을 구했습니다.

만약 -1이 나오면 nums배열에 target이 존재하지 않을 경우 [-1, -1]를 해주었습니다.

 

nums배열에 target이 여러개일 경우 끝 값을 찾아주면 되는데요.

이미 nums에서 오름차순으로 정렬이 되어 있어서 시작 값+1을 해주면서 찾아주면 됩니다.