1. 程式人生 > >LeetCode-922-陣列排序2

LeetCode-922-陣列排序2

大概意思是陣列中索引號為奇數的數字也要為奇數,索引號為偶數的數字也要為偶數,我之前寫的程式碼有問題,先上反例

class Solution {
    public int[] sortArrayByParityII(int[] A) {
        int N = A.length;
        int[] ans = new int[N];
        for(int i = 0;i<N;i = i+2){
//該處只有if語句判斷,容易造成漏判,下面同理
//而且用兩個if語句很麻煩,所以想辦法把條件都寫到一個for迴圈中,一次迴圈結束
            if(A[i]%2 == 0){
                ans[i] = A[i];
            }
        }
         for(int i = 1;i<N;i = i+2){
            if(A[i]%2 == 1){
                ans[i] = A[i];
            }
        }

        return ans;
    }
}

經過修改,以下程式碼

class Solution {
    public int[] sortArrayByParityII(int[] A) {
        int N = A.length;
        int[] ans = new int[N];
        for(int m = 0, n = 1,i = 0;i<N;i++){
            if(A[i]%2  == 0){
                ans[m] = A[i];
                m = m+2;
            }else{
                ans[n] = A[i];
                n = n+2;
            }
        }
        return ans;
    }
}