HDU 5360 Hiking（线段树）

题意：

n个人接受邀请的条件是已经接受邀请的人数区间在l[i] , r[i]

问怎样设置邀请顺序能使得接受邀请的人数最多

解析：

先对区间从小按照右边界从小到大排序，如果右边界相同，再按照左边界从小到大排序。因为右边界越小优先级越高，左边界同理。

如果用优先队列，是算出选择当前人选择哪个区间是最优的。

由于本人不会两个条件的优先队列，所以只能换了一种写法来写。

于是我想到了线段树，逆向思维，利用线段树来维护这个区间选择哪个人是最优的。

选择完这个人之后，将这个人置为无穷大并维护线段树。

并记录下当前人，选择的是哪个区间。

my code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define ls o<<1
#define rs o<<1|1
#define lson ls, L, M
#define rson rs, M+1, R
using namespace std;
const int INF = 0x3f3f3f3f;
const int N = 1e5 + 10;
int n;

struct Node {
    int l, r, id;
} node[N];

int minv[N<<2];

inline void pushUp(int o) {
    minv[o] = min(minv[ls], minv[rs]);
}

void build(int o, int L, int R) {
    if(L == R) {
        minv[o] = L;
        return ;
    }
    int M = (L + R)/2;
    build(lson);
    build(rson);
    pushUp(o);
}

int query(int o, int L, int R, int ql, int qr) {
    if(ql <= L && R <= qr) return minv[o];
    int M = (L + R)/2, ret = INF;
    if(ql <= M) ret = min(ret, query(lson, ql, qr));
    if(qr > M) ret = min(ret, query(rson, ql, qr));
    return ret;
}

void modify(int o, int L, int R, int pos) {
    if(L == R) {
        minv[o] = INF;
        return ;
    }
    int M = (L + R)/2;
    if(pos <= M) modify(lson, pos);
    else modify(rson, pos);
    pushUp(o);
}

int ans[N], res[N];
bool vis[N];

bool cmp(Node a, Node b) {
    if(a.r != b.r)
        return a.r < b.r;
    return a.l < b.l;
}

int main() {
    int T;
    scanf("%d", &T);
    while(T--) {
        scanf("%d", &n);
        for(int i = 1; i <= n; i++) {
            scanf("%d", &node[i].l);
            node[i].id = i;
        }
        for(int i = 1; i <= n; i++)
            scanf("%d", &node[i].r);

        sort(node+1, node+n+1, cmp);
        memset(ans, -1, sizeof(ans));
        memset(vis, false, sizeof(vis));
        build(1, 0, n);
        for(int i = 1; i <= n; i++) {
            int ret = query(1, 0, n, node[i].l, node[i].r);
            if(ret != INF) {
                ans[ret] = node[i].id;
                modify(1, 0, n, ret);
            }
        }

        int tot = 0;
        for(int i = 0; i < n; i++) {
            if(ans[i] != -1) {
                res[tot++] = ans[i];
                vis[ans[i]] = true;
            }else break;
        }

        printf("%d\n", tot);
        for(int i = 1; i <= n; i++) {
            if(!vis[i]) res[tot++] = i;
        }
        for(int i = 0; i < tot; i++) {
            printf("%d", res[i]);
            if(i == tot-1) puts("");
            else putchar(‘ ‘);
        }
    }
    return 0;
}

版权声明：本文为博主原创文章，未经博主允许不得转载。