BZOJ 1303: [CQOI2009]中位数图(思路题)

最新推荐文章于 2018-12-23 15:50:00 发布

weixin_30360497

最新推荐文章于 2018-12-23 15:50:00 发布

阅读量73

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/sdfzsyq/p/10093125.html

本文介绍了一种解决特定算法竞赛问题的方法，通过将原序列转化为1和-1，使用前缀和与计数数组来快速统计答案。代码示例展示了如何实现这一算法，包括输入处理、转换、计数及最终结果输出。

传送门

解题思路

　　比较好想的思路题。首先肯定要把原序列转化一下，大于\(k\)的变成\(1\)，小于\(k\)的变成\(-1\)，然后求一个前缀和，还要用\(cnt[]\)记录一下前缀和每个数出现了几次，然后统计答案的时候从\(1\)循环到\(k\)，每次转化为前缀和相减即可。

代码

#include<iostream>
#include<cstdio>
#include<cstring>

using namespace std;
const int MAXN = 100005;
typedef long long LL;

inline int rd(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {f=ch=='-'?0:1,ch=getchar();}
    while(isdigit(ch)) {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return f?x:-x;
}

int n,m,a[MAXN],p;
int cnt[MAXN<<1],tag;
LL ans;

int main(){
    n=rd(),m=rd();
    for(int i=1;i<=n;i++){
        a[i]=rd();
        if(a[i]>m) a[i]=1;
        else if(a[i]==m) a[i]=0,p=i;
        else a[i]=-1;
        a[i]+=a[i-1];
        if(p) cnt[a[i]+100000]++;
    }
    ans+=cnt[100000];
    for(int i=1;i<p;i++) ans+=cnt[a[i]+100000];
    printf("%lld\n",ans);
    return 0;
}

转载于:https://www.cnblogs.com/sdfzsyq/p/10093125.html