diff --git a/src/main/java/com/cool/ling_cha_mount/sliding_windows/Num1652.java b/src/main/java/com/cool/ling_cha_mount/sliding_windows/Num1652.java new file mode 100644 index 0000000..41f0ec9 --- /dev/null +++ b/src/main/java/com/cool/ling_cha_mount/sliding_windows/Num1652.java @@ -0,0 +1,57 @@ +package com.cool.ling_cha_mount.sliding_windows; + +import org.junit.Test; + +import java.util.Arrays; + +/** + * Created with IntelliJ IDEA. + * + * @Author: Cool + * @Date: 2024/10/07/20:48 + * @Description: + */ +public class Num1652 { + public int[] decrypt(int[] code, int k) { + int[] res = new int[code.length]; + if (k > 0) { + int sum = 0; + int right = 1; + for (int i = 0; i < k; i++) { + sum += code[right++]; + } + for (int i = 0; i < code.length; i++) { + if (right >= code.length) { + right = 0; + } + res[i] = sum; + sum += code[right++]; + if (i != code.length - 1) { + sum -= code[i + 1]; + } + } + } else if (k < 0) { + int right = code.length - 2; + int sum = 0; + k = -k; + for (int i = code.length - 1; i > code.length - 1 - k; i--) { + sum += code[right--]; + } + for (int i = code.length - 1; i >= 0; i--) { + if (right < 0) { + right = code.length - 1; + } + res[i] = sum; + sum += code[right--]; + if (i != 0) { + sum -= code[i - 1]; + } + } + } + return res; + } + @Test + public void test(){ + decrypt(new int[]{5,7,1,4},3); + } +}