본문 바로가기
알고리즘

[알고리즘] Shuffle the Array

by keel_im 2021. 1. 19.
반응형

쉬운 문제를 풀면 자신감이 올라간다. 다음 난이도로 넘어가서 좌절을 겪을 필요가 있다.

🧶문서는 항상 수정 될 수 있습니다. 비판은 환영합니다. 

c++

#include <vector>

using namespace std;


class Solution {
public:
    vector<int> shuffle(vector<int>& nums, int n) {
        vector<int> vc;

        for(int i=0; i<n; i++){
            vc.push_back(nums[i]);
            vc.push_back(nums[i+n]);
        }

        return vc;
    }
};

 파이썬

class Solution:
    def shuffle(self, nums: List[int], n: int) -> List[int]:
        a = nums[:n]
        b = nums[n:]
        list =[]
        for i in range(len(a)):
            list.append(a[i])
            list.append(b[i])
        return list
반응형

댓글