BZOJ 1303: [CQOI2009]中位數圖(思路題)
阿新 • • 發佈:2018-12-09
解題思路
比較好想的思路題。首先肯定要把原序列轉化一下,大於\(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; }