bzoj3192: [JLOI2013]刪除物品(樹狀陣列)

Hanks_o發表於2017-12-09

題目傳送門
好題啊好思維。

解法:
亮哥過來問我這道題。
然後我就開始做了。。
想著想著這個人就過來跟我說把他們並起來就好了。
然後換頂的時候就相當於把分界線移動一格而已。
然後不就樹狀陣列維護就好了。

哦還有就是本來三天前就A了可是沒加longlong硬是WA。

兩個桶頭對頭。
拿樣例來說:
1
4
5
2
7
3
頭對頭。那麼就是541273.
然後分界線在1的後面2的前面。
然後換頭就相當於把分界線移動。
然後樹狀陣列維護就好了呀。

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
struct node {ll x,s;}a[210000],b[110000];int n1,n2,n;
bool cmp(node n1,node n2) {return n1.s>n2.s;}
int lowbit(int x){return x&-x;}ll sum[110000];
void change(int x,int k) {while(x<=n){sum[x]-=k;x+=lowbit(x);}}
ll getsum(int x) {ll ans=0;while(x>0) {ans+=sum[x];x-=lowbit(x);}return ans;}
int main() {
    scanf("%d%d",&n1,&n2);
    for(int i=1;i<=n1;i++)scanf("%lld",&a[n1-i+1].s);
    for(int i=1;i<=n2;i++)scanf("%lld",&a[n1+i].s);n=n1+n2;
    for(int i=1;i<=n;i++)a[i].x=i;sort(a+1,a+1+n,cmp);
    memset(sum,0,sizeof(sum));for(int i=1;i<=n;i++)change(i,1);
    ll f=n1,ans=0;
    for(int i=1;i<=n;i++) {
        if(a[i].x<=f) {ans+=abs(getsum(f)-getsum(a[i].x));f=a[i].x;change(a[i].x,-1);}
        else {ans+=abs(getsum(a[i].x-1)-getsum(f));f=a[i].x;change(a[i].x,-1);}
    }
    printf("%lld\n",ans);
    return 0;
}

相關文章