Crane POJ 2991 （线段树 + 几何数学） 挑战程序设计竞赛170页

POJ原题链接

题意：
有一台起重机。我们把起重机看成由N条线段依次首尾相接而成。第 $i$ 条线段的长度是 $L_i$，最开始，所有的线段都笔直连接，指向上方。
有C条操纵起重机的指令。指令 $i$ 给出两个整数 $S_i$ 和 $A_i$，效果是使线段 $S_i$ 和 $S_{i+1}$ 之间的角度变成 $A_i$度。其中角度指的是从线段 $S_i$ 开始沿逆时针方向旋转到 $S_{i+1}$ 所经过的角度。最开始时所有角度都是180度。
按顺序执行这C条指令。在每条指令执行之后，给出起重机的前端（第N条线段的端点）的坐标。假设起重机的支点的坐标是（0,0）。

限制条件
1 <= N, C <= 10000
1 <= $L_i$ <= 100
1 <= $S_i$ < N, 0 <= $A_i$ <=359

思路
将起重机的每一段看作是一个从坐标原点（0,0）出发的向量，那么最终的答案就抽象为了所有线段向量的累加和，由于需多次修改线段区间的向量方向，那么这个时候再利用线段树维护区间和即可。

注意
由于在修改第 $i$ 个线段向量的方向时，第$i+1，i+2,...N$ 个的线段向量的方向也需要发生改变，因此在线段树修改的时候需要注意如果当前节点左儿子包含修改点，那么当前右儿子的整个区间都需要修改向量方向。
除此之外，还需要注意由于两相邻线段的夹角是相对彼此的，而不是相对于x轴的，因此在计算修改后的向量时，需要按照当前向量角度和修改后的向量角度的相对角度进行修改，这样则可以稳定维护向量变化，保证最后可以从线段树根节点直接输出相对于x轴的正确答案。

附加
（其中的向量按相对变化角度得到的新向量的公式是通过数学几何推出）
设当前向量为 $(x_0,y_0)$，与x轴成 $\alpha$ 角，当前向量通过转动 $\beta$ 角得到新向量 $(x,y)$，设当前向量的模为L（等价于 $\sqrt{x_0^2+y_0^2}$，也是 $\sqrt{x^2+y^2}$），那么可以得到公式
$L\ast cos\alpha =x_0$
$L\ast sin\alpha =y_0$
$L\ast cos(\alpha +\beta )=x$
$L\ast sin(\alpha +\beta )=y$
又因为 $cos(\alpha +\beta )=cos\alpha \ast cos\beta -sin\alpha \ast sin\beta$
$sin(\alpha +\beta )=sin\alpha \ast cos\beta +cos\alpha \ast sin\beta$
综上 $x=x_0\ast cos\beta -y_0\ast sin\beta$
$y=x_0\ast sin\beta +y_0\ast cos\beta$

PS： 本题代码和挑战程序设计竞赛书上172页上的代码略有不同，但是思维一致

AC代码如下：

#include "cstdio"
#include "iostream"
#include "math.h"
#define lson p << 1
#define rson p << 1 | 1
#define int long long
using namespace std;
const int N = 1e4 + 5;
const double PI = acos(-1.0);
int n, c, index, L[N]; ///L数组保存起始线段长度
double prv[N], val;  ///prv数组保存当前第i线段与i+1线段的角度
struct node {
    double x, y, ang; ///分别为向量的坐标，和向量的相对初始变化值
} tree[N << 2];
void build (int p, int l, int r) {
    tree[p].ang = tree[p].x = 0.0;
    if (r == l)
        tree[p].y = L[l];
    else {
        int chl = lson, chr = rson;
        int mid = (l + r) >> 1;
        build(chl, l, mid);
        build(chr, mid + 1, r);
        tree[p].y = tree[chl].y + tree[chr].y;
    }
}
void update (int p, int index, double val, int l ,int r) {
    if (l == r) return; ///叶子结点直接返回
    int chl = lson, chr = rson;
    int mid = (l + r) >> 1;
    if (index <= mid) update(chl, index, val, l, mid);
    if (index > mid) update(chr, index, val, mid + 1, r);
    if (index <= mid) tree[p].ang += val;  ///修改值在左儿子，那么右儿子整个区间都需要修改
    double si = sin(tree[p].ang), co = cos(tree[p].ang);
    tree[p].x = tree[chl].x + (co * tree[chr].x - si * tree[chr].y);///当前节点的向量值由左右儿子的和构成
    tree[p].y = tree[chl].y + (si * tree[chr].x + co * tree[chr].y);
}
signed main () {
#ifndef ONLINE_JUDGE
    freopen("G://Code/in.txt", "r", stdin);
#endif
    int cnt = 0;
    while (~scanf("%lld %lld", &n, &c)) {
        if (cnt++)
            printf("\n");
        for (int i = 1; i <= n; ++i)
            scanf("%lld", &L[i]);
        fill(prv + 1, prv + n, PI); ///初始夹角都为180度
        build(1, 1, n);
        while (c--) {
            scanf("%lld %lf", &index, &val);
            val = val / 180.0 * PI;  ///角度制转弧度制
            update(1, index, val - prv[index], 1, n); ///更新相对值
            prv[index] = val;    ///保存当前角度
            printf("%.2f %.2f\n", tree[1].x, tree[1].y);
        }
    }
    return 0;
}