[LeetCode] 4. Median of Two Sorted Arrays

Chobby·2024년 8월 13일
1

LeetCode

목록 보기
41/194

두 배열을 병합한 후 재정렬하고, 중앙값을 구해주면 된다.

유의해야할 점은 병합된 배열의 길이가 1일 경우 짝수와 홀수를 떠나 해당 요소를 바로 반환하면 된다.

function findMedianSortedArrays(nums1: number[], nums2: number[]): number {
    const concatedArr = nums1.concat(nums2).sort((a, b) => a - b)
    if(concatedArr.length === 1) return concatedArr[0]
    const middleIdx = Math.floor(concatedArr.length / 2)
    if(concatedArr.length % 2 === 0) {
        const sum = concatedArr[middleIdx-1] + concatedArr[middleIdx]

        return sum/2
    } else {
        return concatedArr[middleIdx]
    }
};
profile
내 지식을 공유할 수 있는 대담함

0개의 댓글