6248. 统计中位数为 K 的子数组（力扣）

最新推荐文章于 2024-05-14 20:06:41 发布

我要去看得最远的地方

最新推荐文章于 2024-05-14 20:06:41 发布

阅读量136

点赞数

分类专栏：思维文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/qq_42434171/article/details/128072019

版权

思维专栏收录该内容

20 篇文章 0 订阅

订阅专栏

class Solution {
public:
    int countSubarrays(vector<int>& nums, int k) {
        int id=-1;
        map<int,int>mp;
        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]==k)
            {
                id=i;
                break;
            }
        }
        if(id==-1) return 0;
        int x1=0;
        int x0=0;
        mp[0]=1;
        //mp[x]表示大于k的数减去小于k的数的值
        for(int i=id-1;i>=0;i--)
        {
              if(nums[i]>k) x1++;
              x0=id-i-x1;
              mp[x1-x0]++;
        }
        x1=0;
        x0=0;
        int res=0;
        for(int i=id;i<nums.size();i++)
        {
              if(nums[i]>k)x1++;
              x0=i-id-x1;
              res+=mp[x0-x1]+mp[x0-x1+1];

        }
        return res;
    }
};