平均数为k的最长连续子数组C++实现

平均数为k的最长连续子数组

题目描述：给定n个正整数组成的数组，求平均数正好等于 k 的最长连续子数组的长度。

输入描述：

第一行输入两个正整数n和k，用空格隔开。

第二行输入n个正整数，用来表示数组。

输出描述：

如果不存在任何一个连续子数组的平均数等于k，则输出-1。

否则输出平均数正好等于k的最长连续子数组的长度。

输入例子：

5 2
1 3 2 4 1

输出例子：

3

例子说明：取前三个数即可，平均数为2。

/*
 描述：给定n个正整数组成的数组，求平均数正好等于k的最长连续子数组的长度
 *
 思路:
        1.既然要求平均数为k,只需把所有元素减去k,若一些元素和为0即说明这些元素平均数为k
        2.使用map<sum,i>表示前i个位置的和
        3.遍历元素进行求和。
            当出现不存在的sum时进行记录，记录sum和此时的下标i
            当出现已存在的sum时,假设此时下标为j，表中原始记录的sum下标为i
            则说明在[i+1,j]这一段所有元素的和为0,即其平均数是k,是符合要求的
        4.找出最长的子数组即可
        比如：  1， 3.   2，  4，  5，   1，  1 ， 1，   1  要求平均数为2
        下标：  0， 1，  2，  3，  4，   5，  6，  7，   8
        减2：  -1， 1，  0，  2，  3 ， -1， -1， -1 ， -1

        在下标为3时记录的sum值为2， 在下标为7的时候又为2了，那么在3-7 这个区间内的元素和为0；
        为什么呢，sum由2到2，那么在3-7这个区间里，sum可能先增加上去了，然后有减去了，既然sum不变，那么中间就是0；
 */

#include<iostream>
#include<vector>
#include<unordered_map>

using namespace std;

int main()
{
    int n, k;  //n是长度， k是平均数
    cin >> n >> k;

    vector<int> v(n, 0);
    for (int i = 0; i < n; i++)
    {
        cin >> v[i];
    }

    //定义一个哈希表，key存储sum, val存储下标
    unordered_map<long, int> ump;

    //记录最大的长度
    int maxLen = -1;

    ump[0] = -1;
    long long sum = 0;

    for (int i = 0; i < n; i++)
    {
        sum = sum + v[i] - k;

        if (ump.count(sum))
        {
            maxLen = max(maxLen, i - ump[sum]);
        }
        else
        {
            ump[sum] = i;
        }
    }

    cout<<maxLen<<endl;

    for (unordered_map<long, int> :: iterator it = ump.begin(); it != ump.end(); it++)
    {
        cout<< (*it).first <<" "<<(*it).second<<endl;
    }
    return 0;
}