码迷,mamicode.com
首页 > 其他好文 > 详细

POJ 2481 Cows (线段树)

时间:2014-08-06 23:14:22      阅读:254      评论:0      收藏:0      [点我收藏+]

标签:style   http   color   os   io   strong   for   ar   

Cows


题意:有N头牛,每只牛有一个值[S,E],如果对于牛i和牛j来说,它们的值满足下面的条件则证明牛i比牛j强壮:Si <=Sjand Ej <= Ei and Ei - Si > Ej - Sj。现在已知每一头牛的测验值,要求输出每头牛有几头牛比其强壮。

思路:将牛按照S从小到大排序,S相同按照E从大到小排序,这就保证了排在后面的牛一定不比前面的牛强壮。再按照E值(离散化后)建立一颗线段树(这里最值只有1e5,所以不用离散化也行),遍历每一头牛,在它之前的,E值大于它的牛的数目即是答案(要注意两者相同的情况)。
其实,上面的线段树就是排序好之后的E值序列的中每一个数的逆序对数目。

代码:
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<cmath>
#include<cstdio>
#include<vector>
#include<string>
#include<fstream>
#include<cstring>
#include<ctype.h>
#include<iostream>
#include<algorithm>
#define INF (1<<30)
#define PI acos(-1.0)
#define mem(a, b) memset(a, b, sizeof(a))
#define rep(i, n) for (int i = 0; i < n; i++)
#define debug puts("===============")
typedef long long ll;
using namespace std;
const int maxn = 100100;
int n;
struct node {
    int x, y, id;
}e[maxn];
bool cmp(node s, node v) {
    if (s.x == v.x) return s.y > v.y;
    return s.x < v.x;
}
int x[maxn], index[maxn], dis[maxn];
int discrete(int x[], int index[], int dis[], int n) {
    int cpy[n];
    for (int i = 0; i < n; i++) {
        x[i] = e[i].y;
        cpy[i] = x[i];
    }
    sort(cpy, cpy + n);
    int tot = unique(cpy, cpy + n) - cpy;
    for (int i = 0; i < n; i++) {
        dis[i] = lower_bound(cpy, cpy + tot, x[i]) - cpy;
        index[dis[i]] = i;
    }
    return tot;
}
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1
int has[maxn];
int sum[maxn << 2];
void pushup(int rt) {
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
}
void build(int l, int r, int rt) {
    sum[rt] = 0;
    if (l == r) return ;
    int m = (l + r) >> 1;
    build(lson);
    build(rson);
}
void add(int pos, int x, int l, int r, int rt) {
    if (l == r) {
        sum[rt] += x;
        return ;
    }
    int m = (l + r) >> 1;
    if (pos <= m) add(pos, x, lson);
    else add(pos, x, rson);
    pushup(rt);
}
int query(int L, int R, int l, int r, int rt) {
    if (L <= l && r <= R) return sum[rt];
    int m = (l + r) >> 1;
    int res = 0;
    if (L <= m) res += query(L, R, lson);
    if (R > m) res += query(L, R, rson);
    return res;
}
int main () {
    while(~scanf("%d", &n), n) {
        for (int i = 0; i < n; i++) {
            scanf("%d%d", &e[i].x, &e[i].y);
            e[i].id = i;
        }
        sort(e, e + n, cmp);
        int m = discrete(x, index, dis, n);
        //rep(i, n) cout<<e[i].x<<" "<<e[i].y<<"-------->"<<e[i].id<<" "<<dis[i]<<endl;
        int nowx = -1, nowy = -1, ans = 0, cnt = 1;
        build(0, m - 1, 1);
        for (int i = 0; i < n; i++) {
            if (e[i].x == nowx && e[i].y == nowy) {
                has[e[i].id] = ans;
            } else {
                ans = query(dis[i], m - 1, 0, m - 1, 1);
                has[e[i].id] = ans;
                nowx = e[i].x, nowy = e[i].y;
            }
            add(dis[i], 1, 0, m - 1, 1);
        }
        for (int i = 0; i < n; i++) printf("%d%c", has[i], i == n - 1 ? '\n' : ' ');
    }
    return 0;
}


POJ 2481 Cows (线段树),布布扣,bubuko.com

POJ 2481 Cows (线段树)

标签:style   http   color   os   io   strong   for   ar   

原文地址:http://blog.csdn.net/sio__five/article/details/38405389

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!