20170831美团笔试：长度为n的数组中最长的和为K的倍数的连续子数组

看到这道题，第一个想法是遍历，从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;
}

这里要说的是，我的代码有可以修改的地方，每次改变求和个数时是重新计算的和，其实可以每次从头计算时把结果记下来，然后下次计算用上次的和减去少掉的一个元素，就可以了，复杂度又降低了很多。