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

//活动地址： 牛客春招刷题训练营 - 编程打卡活动
#include<bits/stdc++.h>

using  namespace std;

#define mod 1000000007

typedef long long ll;

#define int  long long

inline ll read() // void int &n

{

    ll s=0,f=1;

    char c=getchar();

    while(c>'9'||c<'0')     

{

        if(c=='-')  

            f=-1;   

        c=getchar();

    }

    while(c>='0'&&c<='9')

    {

        s=(s<<1)+(s<<3)+c-'0';

        c=getchar();

    }

    return s*f;

}

inline void write(int n)

{

if(n<0)

{

  putchar('-');

  n=-n;

}

    

if(n>10) write(n/10);

putchar(n%10+'0');

}

int jiechen(int n)

{

    int sum = 1;

    for (int i = 2; i <= n; i++)

        sum = sum * i % mod;

    return sum % mod;

}

int qsm(ll a, ll p)

{

    ll s=1;

    while(p)

    {

        if(p&1)

         s=s*a%mod;

      a=a*a%mod;

    }

       return s;

}

ll isprime(ll x)

{

    if(x<2)

        return 0;

    for(int i=2;i<=x/i;i++)

        if(x%i==0)

            return 0;

    return 1;
}

bool cmp(int x, int y){
    return x>y;
}

const int N=3e6+10;

const int M =100000;

int a[N];
void solve(){   
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;i++)cin>>a[i];
    map<int,int>m;
    // a[i] 还指代前缀和
     m[0]=0;
    for(int i=1;i<=n;i++){
        a[i]-=k;
        a[i]+=a[i-1];
        // m.count() 和 m[] 有区别的
        if(m.count(a[i]))continue;
        m[a[i]]=i; // 记录a[i] 第一次出现的下标
    }
    int ans=-1;
    for(int i=1;i<=n;i++){
        if(m.count(a[i])&&i>m[a[i]])ans=max(ans,i-m[a[i]]);
    }
    cout<<ans;
}

signed main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
    int _=1;
    //cin>>_;
    while(_--)
    {
        solve();
    }
    return 0;
}