LeetCode 1652. Defuse the Bomb

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

題目大意

拆除炸彈遊戲

給定長度為 n 的環狀陣列 code 和密鑰 k

• 如果 k > 0 將第 i 個數字替換為後 k 個數字的總和
• 如果 k < 0 將第 i 個數字替換為前 k 個數字的總和
• 如果 k == 0 將第 i 個數字替換為 0

思考

依照題意即可

C++ 參考答案：

class Solution
{
public:
    vector<int> decrypt(vector<int> &code, int k)
    {
        const int n = code.size();
        vector<int> ans(n);

        if (k == 0)
            return ans;

        int sign = 1;
        if (k < 0)
        {
            k *= -1;
            sign = -1;
        }

        for (int i = 0; i < n; ++i)
        {
            for (int j = 1; j <= k; ++j)
            {
                ans[i] += code[(i + j * sign + n) % n];
            }
        }

        return ans;
    }
};

Go 參考答案：

func decrypt(code []int, k int) []int {
	n := len(code)
	ans := make([]int, n)

	if k == 0 {
		return ans
	}

	sign := 1
	if k < 0 {
		k = -k
		sign = -1
	}

	for i := 0; i < n; i++ {
		for j := 1; j <= k; j++ {
			ans[i] += code[(i+j*sign+n)%n]
		}
	}

	return ans
}