看到这道题,第一个想法是遍历,从n个数求和,再到n-1个数求和(这里需要遍历)。再到1个数,没有k的倍数则输出0。但是一看数量级10^5,O(n^2)的复杂度肯定超了,于是想到了以前项目里用到的滑窗法,暂且这么叫吧。当时老师让做图像积分的同学做的。借鉴下。就是m个数的和,可以求出前m个,在减去最开始的,加上最后面数下一个就是下一个积分和。看图:
具体代码如下:
#include <iostream> using namespace std; int resolution(int *p, int n, int k) { int sum = 0; for (int i = 0; i < n; i++) { sum += p[i]; } if (sum%k == 0) { return n; } for (int i = n-1; i > 0; i--) { sum = 0; for (int ii = 0; ii < i; ii++) { sum += p[ii]; } if (sum%k == 0) return i; for (int j = 0; j < n-i; j++) { sum = sum - p[j] + p[j + i]; if (sum%k == 0) return i; } } return 0; } int main() { int n, k; while (cin >> n) { int *p = new int[n]; for (int i = 0; i < n; i++) { cin >> p[i]; } cin >> k; cout << resolution(p, n, k) << endl; delete[] p; } return 0; }这里要说的是,我的代码有可以修改的地方,每次改变求和个数时是重新计算的和,其实可以每次从头计算时把结果记下来,然后下次计算用上次的和减去少掉的一个元素,就可以了,复杂度又降低了很多。