1. 程式人生 > >[BZOJ4240]有趣的家庭菜園(貪心+樹狀陣列)

[BZOJ4240]有趣的家庭菜園(貪心+樹狀陣列)

最後數列一定是單峰的,問題就是最小化最後的位置序列的逆序對數。

從大到小加數,每次貪心看放左邊和右邊哪個產生的逆序對數更少,樹狀陣列即可。

由於大數放哪對小數不產生影響,所以正確性顯然。

注意相同數之間一定能不構成逆序對,需要特判。

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 #define rep(i,l,r) for (int i=(l); i<=(r); i++)
 5 typedef long long ll;
 6 using namespace
std; 7 8 const int N=300010; 9 ll ans; 10 int n,c[N],a[N],p[N]; 11 12 bool cmp(int x,int y){ return a[x]>a[y]; } 13 void add(int x,int k){ for (; x<=n; x+=x&-x) c[x]+=k; } 14 int que(int x){ int res=0; for (; x; x-=x&-x) res+=c[x]; return res; } 15 16 int main(){ 17 freopen("
bzoj4240.in","r",stdin); 18 freopen("bzoj4240.out","w",stdout); 19 scanf("%d",&n); 20 rep(i,1,n) scanf("%d",&a[i]),p[i]=i; 21 sort(p+1,p+n+1,cmp); 22 for (int i=1; i<=n; ){ 23 int j=i; 24 for (; j<=n; j++){ 25 int x=que(p[j]); ans+=min(x,i-1
-x); 26 if (a[p[j]]!=a[p[j+1]]) break; 27 } 28 for (; i<=j; i++) add(p[i],1); 29 } 30 printf("%lld\n",ans); 31 return 0; 32 }