[리트코드/JAVA] 1470. Shuffle the Array

 

leetcode.com/problems/shuffle-the-array/

Shuffle the Array - LeetCode

 

# 문제설명

Given the array nums consisting of 2n elements in the form [x1,x2,...,xn,y1,y2,...,yn].

Return the array in the form [x1,y1,x2,y2,...,xn,yn].

 

# 제한사항

• 1 <= n <= 500
• nums.length == 2n
• 1 <= nums[i] <= 10^3

# 입출력 예

Example 1:

Input: nums = [2,5,1,3,4,7], n = 3

Output: [2,3,5,4,1,7]

Explanation: Since x1=2, x2=5, x3=1, y1=3, y2=4, y3=7 then the answer is [2,3,5,4,1,7].

 

Example 2:

Input: nums = [1,2,3,4,4,3,2,1], n = 4

Output: [1,4,2,3,3,2,4,1]

 

Example 3:

Input: nums = [1,1,2,2], n = 2

Output: [1,2,1,2]

# 풀이

class Solution {
    public int[] shuffle(int[] nums, int n) {
        int[] output = new int[nums.length];
        
        int idx = 0;
        for(int i=0; i<n; i++) {
        	output[idx] = nums[i];
        	idx+=2;
        }
        
        int idx2 = 1;
        for(int i=n; i<nums.length; i++) {
        	output[idx2] = nums[i];
        	idx2 += 2;
        }
        
        return output;
    }
}

 

# 3

리턴할 배열을 nums의 길이와 동일하게 선언합니다.

 

#5, 11

n을 기준으로 나눠서 값을 섞어야 해서 인덱스로 사용할 변수 2개를 선언합니다.

첫 번째 idx는 0, 두 번째 idx2는 1로 선언합니다. 배열의 인덱스가 [0, 1, 2, 3, ... .] 과 같은 형태이기 때문입니다.

idx와 idx2를 다음과 같이 활용할 것입니다. [idx0, idx1, idx2, idx3, ... .]

 

# 6

첫 번째 for문은 n까지 범위를 정합니다. n을 기준으로 반으로 나누기 때문입니다.

예시 1번으로 보면 [2,5,1,3,4,7] 배열 중에서 [2, 5, 1]까지만 새로운 배열 output에 넣어줄 겁니다.

 

# 7 ~ 8

output 배열에 넣어줄 때 idx를 +2씩 더해서 2, 5, 1을 [2, x, 5, x, 1, x] 처럼 넣어줍니다.

 

# 12 ~ 15

아래는 위와 동일한 방식입니다. 대신 i는 n으로 선언하고 범위는 nums배열의 크기만큼 선언합니다.