[LeetCode/Easy] 1652. Defuse the Bomb

1. Input

1) int[] code

- code.length == n

2) int k

 

2. Output

1) code를 풀기 위해 다음과 같은 단계를 거치고 난 후 암호가 해제된 결과 배열을 반환

- k>0: code[i] = i번째의 다음 k개의 합

- k<0: code[i] = i번째의 이전 k개의 합

- k=0: code[i] = 0

- code는 순환 배열이다. 즉 code[n-1] 다음은 code[0]이고, code[0]의 이전은 code[n-1]이다.

 

3. Constraint

1) 1 <= n <= 100

2) 1 <= code[i] <= 100

3) -(n-1) <= k <= n-1

 

4. Example

Input: code = [5,7,1,4], k = 3 -> Output: [12,10,16,13]

설명: [7+1+4, 1+4+5, 4+5+7, 5+7+1]

 

5. Code

1) 첫 코드(2023/04/15)

int[] answer = new int[code.length];
if(k>0){
    for(int i=0 ; i<answer.length ; i++)
        for(int j=1 ; j<=k ; j++)
            answer[i] += code[(i+j)%code.length];
} else if(k<0){
    for(int i=0 ; i<answer.length ; i++)
        for(int j=1 ; j<=-k ; j++){
            if(i-j<0) answer[i] += code[(i-j+code.length)%code.length];
            else answer[i] += code[(i-j)%code.length];
        }
}
return answer;