HDU 4302 Holedox Eating （二分+樹狀數組維護）

題意：一個小孩吃蛋糕，他的起始點在0位置，現在有m個操作，0 x 代表在x位置出現一個蛋糕；  1 代表小孩要去離他最近的點吃蛋糕，如果與他距離最小的點有兩個，則去與他上次走的方向相同的點； 如果沒有蛋糕，他就不動。 詢問m次操作過後，小孩走了多少距離。

每一次，用二分判斷，他右邊離他最近的蛋糕點，和左邊離他最近的蛋糕點。如何判斷呢，只需用樹狀數組維護蛋糕總數，判斷小孩當前位置pos 與 mid之間的蛋糕數是否大於等1，是的話可以嘗試逼近，否則就遠離。

需要注意線段是1 --- L ，而小孩起始點在0，所以有L + 1 個點。

#include <iostream>
#include <algorithm>
#include <cmath>
#include<functional>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <vector>
#include <set>
#include <queue>
#include <stack>
#include <climits>//形如INT_MAX一類的
#define MAX 100050
#define INF 0x7FFFFFFF
using namespace std;

int n,m;
int c[MAX];

int lowbit(int x) {
    return x & (-x);
}

void update(int x,int va) {
    while(x < n + 2) {
        c[x] += va;
        x += lowbit(x);
    }
}

int query(int x) {
    int sum = 0;
    while(x > 0) {
        sum += c[x];
        x -= lowbit(x);
    }
    return sum;
}

int judge(int p,int rp,int lp,int flag) {
    if(rp == -1 && lp == -1) return -1;
    if(rp == -1) return 0;
    if(lp == -1) return 1;
    if(rp - p > p - lp) return 0;
    if(rp - p < p - lp) return 1;
    if(flag == 0) return 0;
    return 1;
}

int find(int pos,int l,int r,int kind) {
    int mid , des;
    if(kind == 1) des = INF;
    else des = -1;
    while(l <= r) {
        mid = (l + r) >> 1;
        if(kind == 1) {
            if(query(mid) - query(pos - 1) >= 1) {
                des = min(des,mid);
                r = mid - 1;
            } else l = mid + 1;
        } else {
            if(query(pos) - query(mid - 1) >= 1) {
                des = max(des,mid);
                l = mid + 1;
            } else r = mid - 1;
        }
//        cout << l << ' ' << mid << ' ' << r << endl;
    }
    if(des == INF) return -1;
    return des;
}

int main() {
    int T;
    cin >> T;
    int ca = 1;
    while(T--) {
        scanf("%d%d",&n,&m);
        memset(c,0,sizeof(c));
        int a,b;
        int pos = 1, flag = 1; //flag 1:right  0: left
        __int64 dis = 0;
        for(int i=0; i<m; i++) {
            scanf("%d",&a);
            if(a == 1) {
                int rpos = find(pos,pos,n+1,1);
                int lpos = find(pos,1,pos,0);
                int dir = judge(pos,rpos,lpos,flag);
                if(dir == -1) continue;
                if(dir == 1) {
                    dis += rpos - pos;
                    pos = rpos;
                    flag = 1;
                } else {
                    dis += pos - lpos;
                    pos = lpos;
                    flag = 0;
                }
                update(pos,-1);
            } else {
                scanf("%d",&b);
                update(b+1,1);
            }
        }
        printf("Case %d: %d\n",ca ++,dis);
    }
    return 0;
}