CF1005E1 Median on Segments (Permutations Edition) 思维

最新推荐文章于 2020-05-12 22:05:24 发布

原创最新推荐文章于 2020-05-12 22:05:24 发布 · 273 阅读

0 ·

CC 4.0 BY-SA版权

综合专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一种算法，用于计算给定数组中以特定数值作为中位数的子区间数量。通过使用前缀和与哈希映射，算法能够高效地找到满足条件的子区间，特别适用于数据规模较大或需要频繁查询的场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

$题目链接：点击打开链接$

$题意：给你n个数和m，问在这n个数中以m为中位数的区间有多少个？$

$因为要使m为中位数，肯定是m的值位于这些数的中间的部分，即要有比m大的数和比m小的数，且大的数和小的数要相等或者大的数比小的数多一$

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <vector>
#include <map>
using namespace std;

int main()
{
    ios::sync_with_stdio(false);
    int n,m;
    cin >> n>>m;
    vector<int>p(n);
    for(int i=0; i<n; i++)
        cin>>p[i];
    map<int,int>mp;
    mp[0]=1;
    bool has=false;
    int sum=0;
    long long ans=0;
    for(int r=0; r<n; r++)
    {
        if(p[r]<m)
            sum--;
        else if(p[r]>m)
            sum++;
        if(p[r]==m)
            has=true;
        if(has)
            ans+=mp[sum]+mp[sum-1];
        else
            mp[sum]++;
    }
    cout<<ans<<endl;
    return 0;
}