LeetCode简单题1652.拆炸弹

最新推荐文章于 2024-11-15 16:27:37 发布

光谷梁朝伟

最新推荐文章于 2024-11-15 16:27:37 发布

阅读量119

点赞数

文章标签： leetcode 算法 java

本文链接：https://blog.csdn.net/Leung_Chiu_Wai/article/details/127032974

版权

class Solution {
    public int[] decrypt(int[] code, int k) {
        int[] res=new int[code.length];
        if(k>0){
            for (int i = 0; i < res.length; i++) {
                for (int j = i+1; j < i+1+k; j++) {
                    if(j<res.length){
                        res[i]+=code[j];
                    }else{
                        res[i]+=code[j-code.length];
                    }
                }
            }
        }else if(k<0){
            for (int i = 0; i < res.length; i++) {
                for (int j = i-1; j>i-1+k; j--) {
                    if(j>=0){
                        res[i]+=code[j];
                    }else{
                        res[i]+=code[code.length+j];
                    }
                }
            }
        }else{
            res=new int[code.length];
        }
        return res;
    }
}