본문 바로가기

Algorithm/Implementation

(C++) - LeetCode (easy) 1652. Defuse the Bomb

반응형

https://leetcode.com/problems/defuse-the-bomb/description/

간단 구현 문제였습니다.

📕 풀이방법

📔 입력 및 초기화

정답 vector ans선언 후 code.size만큼 공간을 할당합니다.

📔 풀이과정

code의 원소를 수행하며 k가 양수인지 음수인지에 따라 k만큼 이동해 누적합을 sum에 저장후 ans를 sum값으로 갱신합니다.

📔 정답 출력 | 반환

ans를 반환합니다.


📕 Code

📔 C++

class Solution {
public:
    vector<int> decrypt(vector<int>& code, int k) {
        vector <int> ans(code.size());
        for(int i = 0; i < code.size(); i++) {
            int sum = 0;
            for (int next = 0; next < abs(k); next++) {
                if(k > 0) {
                    sum += code[(i+1+next) % code.size()];
                } else if (k < 0) {
                    sum += code[(i-1 - next + code.size()) % code.size()];
                }
            }
            ans[i] = sum;
        }
        return ans;
    }
};

*더 나은 내용을 위한 지적, 조언은 언제나 환영합니다.