leetcode 4번문제 Median of Two Sorted Arrays

문제와 예시는 주소에서 확인

https://leetcode.com/problems/median-of-two-sorted-arrays/

Median of Two Sorted Arrays - LeetCode

 

class Solution {
    public double findMedianSortedArrays(int[] a, int[] b) {
        
       int allSize = (a.length + b.length);
 
        int chkSize = (allSize / 2)+1;
        int[] c = new int[chkSize];
        int chkNum = 0;
        int chkNum2 = 0;
        while (chkSize > 0) {
 
            if (a.length - 1 < chkNum) {
                c[chkSize - 1] = b[chkNum2];
                chkNum2++;
                chkSize--;
                continue;
            }
            if (b.length - 1 < chkNum2) {
                c[chkSize - 1]= a[chkNum];
                chkNum++;
                chkSize--;
                continue;
            }
 
            if (a[chkNum] < b[chkNum2]) {
                c[chkSize - 1] = a[chkNum];
                chkNum++;
            } else {
                c[chkSize - 1] = b[chkNum2];
                chkNum2++;
            }
            chkSize--;
        }
        
        if (allSize % 2 == 0) {
            return ((double)(c[0] + c[1])/2);
        }
        
        return (double)c[0];
    }
}