首页 > 代码库 > ZOJ--3574--Under Attack II【线段树+欧拉公式】

ZOJ--3574--Under Attack II【线段树+欧拉公式】

链接:http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3574

题意:一个坐标系,给出x1、x2限定左右边界,有n条直线,告诉每条直线的k和b,问在x1、x2区间内空间被直线分割成几部分


思路:

这道题是比赛时做的,AC之后发现别人都是用归并排序求逆序对数来解的。


说我的解法吧,首先拿到题的时候发现是划分区域这样的,第一下就想到了欧拉公式,但是n有30000,n^2找交点肯定要超时。土豪在纸上画了一下,一条直线必然在x1和x2处留下两点,如果一个直线左端点在另一个直线左端点上面,右端点在另一条线右端点下面,则他们必有一个交点。于是可以转换成线段树来做,给左端点从小到大排序,查询比当前直线右端点大的点的数目,更新欧拉公式要用到的点数和边数,注意交点在x1、x2时的情况。由于欧拉公式适用于连通图,所以要借助x1、x2两条边界线,但是直线没有端点,再虚拟出上下两个边界构出顶点,这样必定是连通图,所以初始时点数和边数都是4。

由于题目说了不会有三条线交于一点的情况,所以处理起来方便很多。


细节:

1.k*x1+b之后的数可能会很大,但是直线最多只有30000条,所以离散化处理。
2.我两个地方写逗了卡了较长时间。写下来防止以后犯逗:

(1)建树顺手写在了输入直线数之后,简单的样例发现不了,因为直线右端点如果没重合的话是不会有影响的。应当写在离散化之后。范围就到离散化后最大的数。

(2)调用query2函数是L、R的取值问题,我为了图省事R直接取到xx+1,如果前一组样例离散化后值较大,后一组较小,则xx+1就会取到上一组的情况,导致答案出错,我尝试建树时范围更大些,但是结果还是WA。解决方案是不要图省事,就取到xx。或者每组样例memset一下sum数组,但是这样显然很慢。


#include<cstring>
#include<string>
#include<sstream>
#include<fstream>
#include<iostream>
#include<iomanip>
#include<cstdio>
#include<cctype>
#include<algorithm>
#include<queue>
#include<deque>
#include<map>
#include<set>
#include<vector>
#include<stack>
#include<ctime>
#include<cstdlib>
#include<functional>
#include<cmath>
using namespace std;
#define PI acos(-1.0)
#define MAXN 30010
#define eps 1e-7
#define INF 0x3F3F3F3F      //0x7FFFFFFF
#define LLINF 0x7FFFFFFFFFFFFFFF
#define seed 1313131
#define MOD 1000000007
#define ll long long
#define ull unsigned ll
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

struct node{
    int y1, y2;
}line[MAXN];
int sum[MAXN << 2];
int n;
map<int, int> mp;
bool cmp(node x, node y){
    return x.y1 < y.y1;
}
bool operator == (node p1, node p2){
    return p1.y1 == p2.y1 && p1.y2 == p2.y2;
}
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 update(int pos, int l, int r, int rt){
    if(l == r){
        sum[rt]++;
        return ;
    }
    int m = (l + r) >> 1;
    if(pos <= m)    update(pos, lson);
    else    update(pos, rson);
    pushup(rt);
}
int query1(int pos, int l, int r, int rt){
    if(l == r){
        return sum[rt];
    }
    int m = (l + r) >> 1;
    if(pos <= m)    query1(pos, lson);
    else    query1(pos, rson);
}
int query2(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 += query2(L, R, lson);
    if(R > m)   res += query2(L, R, rson);
    return res;
}
int aaa[MAXN];
int fuck[MAXN];
int main(){
    int x1, x2;
    int i, j, k, b;
    ll ppp, lll;
    while(scanf("%d%d", &x1, &x2) != EOF){
        mp.clear();
        ppp = 4;
        lll = 4;
        scanf("%d", &n);
        int m = 0;
        int flag = 0;
        for(i = 0; i < n; i++){
            scanf("%d%d", &k, &b);
            line[m].y1 = k * x1 + b;
            line[m].y2 = k * x2 + b;
            if(line[m].y1 != line[m].y2)    flag = 1;
            aaa[m] = line[m].y2;
            m++;
        }
        sort(aaa, aaa + m);
        int xx = unique(aaa, aaa + m) - aaa;
        if(x1 == x2 && flag == 0){
            printf("%d\n", xx + 1);
            continue;
        }
        for(i = 0; i < xx; i++){
            mp[aaa[i]] = i + 1;
        }
        build(1, xx + 1, 1);
        sort(line, line + m, cmp);
        m = unique(line, line + m) - line;
//        cout<<m<<endl;
        ll ans = 0;
        for(i = 0; i < m; i++){
            j = i;
            while(line[i + 1].y1 == line[i].y1 && i + 1 < m)    i++;
            ppp++;
            lll += i - j + 1;
            lll++;
            for(int ii = j; ii <= i; ii++){
                int ttt = mp[line[ii].y2];
                int temp = query1(ttt, 1, n, 1);
                if(!temp){
                    lll++;
                    ppp++;
                }
                if(ttt + 1 <= xx){
                    temp = query2(ttt + 1, xx, 1, n, 1);
                    ppp += temp;
                    lll += temp * 2;
                }
            }
            for(int ii = j; ii <= i; ii++){
                update(mp[line[ii].y2], 1, n, 1);
            }
        }
//        cout<<ppp<<" "<<lll<<endl;
        ans = 2 - ppp + lll - 1;
        printf("%lld\n", ans);
    }
    return 0;
}


ZOJ--3574--Under Attack II【线段树+欧拉公式】