roma-dxunvrs
23 lines
1.5 KiB
Markdown
23 lines
1.5 KiB
Markdown
Leetcode #532 | #Medium | [Префикс](../../Методы/Префикс.md) | [Хэш-таблицы](../../Структуры/Хэш-таблицы.md) | [Математика](../../Методы/Математика.md)
|
|
## Идея
|
|
Любимая идея префиксных сумм. Сумам любого подмассива это разность префиксных сумм - значит они должны давать одинаковый остаток при делении на k. Считаем очередную преф сумму, если ее нет в мапе кладем ее и индекс (получится что запишется минимальный индекс, это нужно чтобы максимизировать длину чтобы условие точно выполнилось), а если такой остаток уже есть, то смотрим на длину подмассива и если >=2 под условие, то возвращаем true.
|
|
## Big-O
|
|
- Время ```O(N)```
|
|
- Память ```O(K)```
|
|
## Код
|
|
```Java
|
|
class Solution {
|
|
public boolean checkSubarraySum(int[] nums, int k) {
|
|
Map<Integer, Integer> map = new HashMap<>();
|
|
map.put(0, -1);
|
|
int pref = 0;
|
|
for (int i = 0; i < nums.length; i++) {
|
|
pref += nums[i];
|
|
int rem = pref % k;
|
|
if (!map.containsKey(rem)) map.put(rem, i);
|
|
else if (i - map.get(rem) >= 2) return true;
|
|
}
|
|
return false;
|
|
}
|
|
}
|
|
``` |