[BZOJ4240]有趣的家庭菜园(贪心+树状数组)

传送门

因为草两边如果比他要高的话草就会死,那么我们肯定要让最高的放中间,然后从最高的开始向两边递减,这样就最优。我马上想到了之前做过的题。

poj2299:
给定一个长度为n的序列,如果只允许进行比较和交换相邻两个数的操作,求至少需要多少次交换才能把序列从小到大排序。

这道题模拟的是冒泡排序,因为每次排一遍序就会使得逆序对少一个,实际上就是求序列中的逆序对。
那么带入这道题,不就是求逆序对嘛,首先按高度从大到小排序,然后依次在左右两边,贪心选择可以形成较少的逆序对的那边放。
注意判断相等的情况。

#include
#include
#include
#include
#include
using namespace std;
typedef long long ll;
const int maxn=300010;
int c[maxn]; int n;
int ls[maxn]; 
int lowbit(int x){return x&-x;}
void add(int x,int v){while(x<=n){c[x]+=v; x+=lowbit(x);}}
int getsum(int x){int sum=0; while(x){sum+=c[x]; x-=lowbit(x);} return sum;}
int val[maxn],pos[maxn];
bool cmp(int a,int b){return val[a]>val[b];}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) {scanf("%d",&val[i]); pos[i]=i;}
    sort(pos+1,pos+n+1,cmp); 
    ll ans=0;int t;
    for(int i=1;i<=n;i=t+1)
    {
        for(int j=i;j<=n;j++)
        {
            int s=getsum(pos[j]);
            ans+=ll(min(s,i-s-1));
            t=j; if(val[pos[j]]!=val[pos[j+1]]) break;
        }
        for(int j=i;j<=t;j++) add(pos[j],1);
    }
    printf("%lld\n",ans);
    return 0;   
}

最后补充一下归并排序求逆序对的做法:
  我们知道归并排序可以将两个有序数列合在一起,比如说:
  [3,4,5],[1,2,6]
  一开始的左指针在3处,右指针在1处,这时候3是比1大的,那么3,4,5都可以和1形成逆序对(注意1在原序列中的位置也会在3,4,5之后)所以可以形成(3,1),(4,1),(5,1)三个逆序对,也就是mid-l+1个逆序对(mid=3,l=1)

你可能感兴趣的:(树状数组,贪心)