POJ 2991 线段树

POJ 2991
题目链接:
http://poj.org/problem?id=2991
题意:
给n个初始线段,线段首尾相连。刚开始它们都在y轴上。
现在每次做旋转操作。操作时,前u个点不动,第u个点相对u-1点做旋转,u点后面的点相对u的位置也不动。
思路:
线段树,区间更新。
因为每次操作相当于更新u及u后面的所有点,所以保存一个点结构,结构里有x、y、angle、flag四个值,分别表示横坐标、纵坐标、角度、延迟标记。更新方式比较奇特,具体见push_down,push_up,func函数,不过也是一种延迟标记思想。
源码:

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <string>
#include <algorithm>
#include <iostream>
using namespace std;
const int MAXN = 10000 + 5;
const double PI = acos(-1.0);
int n, q;
double x[MAXN * 4], y[MAXN * 4];
int angle[MAXN * 4];
int flag[MAXN * 4];
int a[MAXN * 4];
void func(int o, int ang)   ///此处写反 { double tx = x[o], ty = y[o]; x[o] = tx * cos(ang * PI / 180.0) - ty * sin(ang * PI / 180.0);
    y[o] = tx * sin(ang * PI / 180.0) + ty * cos(ang * PI / 180.0);
    angle[o] = (angle[o] + ang) % 360;
    flag[o] = (flag[o] + ang) % 360;
}
void push_up(int o)
{
    x[o] = x[o << 1] + x[(o << 1) | 1];
    y[o] = y[o << 1] + y[(o << 1) | 1];
}
void push_down(int o)
{
     if(flag[o]){
        func(o << 1, flag[o]);
        func((o << 1) | 1, flag[o]);
        flag[o] = 0;
     }
}
void build(int l, int r, int o)
{
    x[o] = 0;
    angle[o] = flag[o] = 0;
    if(l == r){
        y[o] = a[l];
    }
    else{
        int mid = (l + r) >> 1;
        build(l, mid, o << 1);
        build(mid + 1, r, (o << 1) | 1);
        push_up(o);
    }
}
int query(int u, int l, int r, int o)
{
//    printf("u = %d, l = %d, r = %d, o = %d, angle = %d, flag = %d\n", u, l, r, o, angle[o], flag[o]);
    if(l == r) return angle[o];
    else{
        push_down(o);
        int mid = (l + r) >> 1;
        int ang;
        if(u <= mid) ang = query(u, l, mid, o << 1);
        else ang = query(u, mid + 1, r, (o << 1) | 1);
        push_up(o);
        return ang;
    }
}
void update(int l, int r, int ang, int L, int R, int o)
{
//    printf("o = %d, l = %d, r = %d, L = %d, R = %d, ang = %d\n", o, l, r, L, R, ang);
    if(l <= L && r >= R){
        func(o, ang);
    }
    else{
        push_down(o);
        int mid = (L + R) >> 1;
        if(l <= mid) update(l, r, ang, L, mid, o << 1);
        if(mid < r) update(l, r, ang, mid + 1, R, (o << 1) | 1);
        push_up(o);
    }
//    printf("o = %d, angle[o] = %d, flag[o] = %d\n", o, angle[o], flag[o]);
}
int main()
{
//    freopen("data.txt", "r", stdin);
    int f = 1;
    while(scanf("%d%d", &n, &q) != EOF){
        for(int i = 1 ; i <= n ; i++) scanf("%d", a + i);
        if(f) f = 0;
        else printf("\n");
        build(1, n, 1);
//        printf("x[1] = %.2f y[1] = %.2f\n", x[1], y[1]);
        while(q--){
            int u, v;
            scanf("%d%d", &u, &v);
            int a1 = query(u, 1, n, 1);
            int a2 = query(u + 1, 1, n, 1);
//            int ang = query(u, 1, n, 1) - 180 + v - query(u + 1, 1, n, 1);
            int ang = (v - (a2 - a1 + 180 + 360) % 360) % 360;
//            printf("ang = %d, a1 = %d, a2 = %d\n", ang, a1, a2);
//            ang = (ang + 360) % 360;
            update(u + 1, n, ang, 1, n, 1);
            printf("%.2f %.2f\n", x[1], y[1]);
        }
    }
    return 0;
}

你可能感兴趣的:(POJ 2991 线段树)