leetcode 974. 和可被 K 整除的子数组
题目详情
题目链接
给定一个整数数组 A,返回其中元素之和可被 K 整除的(连续、非空)子数组的数目。
- 示例:
输入:A = [4,5,0,-2,-3,1], K = 5
输出:7
解释:
有 7 个子数组满足其元素之和可被 K = 5 整除:
[4, 5, 0, -2, -3, 1], [5], [5, 0], [5, 0, -2, -3], [0], [0, -2, -3], [-2, -3]提示:
- 1 <= A.length <= 30000
- -10000 <= A[i] <= 10000
- 2 <= K <= 10000
我的代码
class Solution {
public:
int subarraysDivByK(vector<int>& A, int K) {
int len = A.size();
int res = 0;
vector<int> results(len, 0);
for (int i = len - 1; i >= 0; --i) {
int sum = 0;
for (int j = i; j < len; ++j) {
sum += A[j];
if (sum % K) {
results[i] += 0;
} else {
results[i] += 1 + ((j == len - 1)? 0: results[j + 1]);
break;
}
}
res += results[i];
}
return res;
}
};
我的成绩
执行结果:超出时间限制
一些想法
本道题我使用了相当于暴力的方法,但是超出了时间限制,差最后一个测试用例没能通过。。。
执行用时为 36 ms 的范例
class Solution {
public:
int subarraysDivByK(vector<int>& A, int K) {
int N = A.size();
if (N < 1 || K < 2) return 0;
int ans = 0;
int sums[N + 1];
int counts[K];
memset(sums, 0, sizeof(sums));
memset(counts, 0, sizeof(counts));
sums[0] = 0;
for (int i = 1; i < N + 1; ++i) {
sums[i] = sums[i - 1] + A[i - 1];
}
for (int i = 0; i < N + 1; ++i) {
counts[(sums[i] % K + K) % K]++;
}
for (int i = 0; i < K; ++i) {
if (counts[i] > 1) {
ans += counts[i] * (counts[i] - 1) / 2;
}
}
return ans;
}
};
思考
参见官方解答