結果

問題 No.1226 I hate Robot Arms
コンテスト
ユーザー chielo
提出日時 2020-09-11 23:18:04
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 235 ms / 2,000 ms
コード長 3,760 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,137 ms
コンパイル使用メモリ 201,572 KB
最終ジャッジ日時 2025-01-14 11:42:38
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 28
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:121:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  121 |     scanf("%d%d", &n, &q);
      |     ~~~~~^~~~~~~~~~~~~~~~
main.cpp:128:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  128 |         scanf("%d%d", &op, &k);
      |         ~~~~~^~~~~~~~~~~~~~~~~
main.cpp:130:18: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  130 |             scanf("%d", &v);
      |             ~~~~~^~~~~~~~~~
main.cpp:134:18: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  134 |             scanf("%d", &v);
      |             ~~~~~^~~~~~~~~~

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>

const int maxn = 1e5 + 3, lo = 0x2'0000;
const double pi = acos(-1);

struct SEG {
    struct node {
        double a[2][2], b[2];
        bool m;
        void tag(double u[2][2], double v[2]) {
            double c[2][2];
            for(int i = 0; i < 2; ++i) {
                for(int j = 0; j < 2; ++j) {
                    c[i][j] = 0.0;
                    for(int k = 0; k < 2; ++k) {
                        c[i][j] += a[i][k] * u[k][j];
                    }
                }
            }
            memcpy(a, c, sizeof(a));
            double d[2];
            for(int i = 0; i < 2; ++i) {
                d[i] = 0;
                for(int k = 0; k < 2; ++k)
                    d[i] += b[k] * u[k][i];
                d[i] += v[i];
            }
            memcpy(b, d, sizeof(b));
            m = true;
        }
        void down(node &u, node &v) {
            if(m) {
                u.tag(a, b);
                v.tag(a, b);
                *this = {{{1.0, 0.0}, {0.0, 1.0}}, {0.0, 0.0}, false};
            }
        }
    } seg[lo * 2];
    void init() {
        for(int i = 0; i < lo * 2; ++i) {
            seg[i] = {{{1.0, 0.0}, {0.0, 1.0}}, {0.0, 0.0}, false};
        }
    }
    void modify(int l, int r, double u[2][2], double v[2]) {
        down(l - 1); down(r);
        for(int i = l + lo, j = r + lo; i < j; i /= 2, j /= 2) {
            if(i & 1) {
                seg[i++].tag(u, v);
            }
            if(j & 1) {
                seg[j++].tag(u, v);
            }
        }
    }
    void down(int k) {
        if(k < 0 || k >= lo)
            return;
        for(int i = 1, j = lo; i < lo; j /= 2, i = i * 2 + bool(k & j)) {
            seg[i].down(seg[i * 2], seg[i * 2 + 1]);
        }
    }
    std::pair<double, double> query(int k, double v[2]) {
        down(k);
        double d[2];
        auto &&u = seg[k + lo];
        for(int i = 0; i < 2; ++i) {
            d[i] = 0;
            for(int j = 0; j < 2; ++j)
                d[i] += v[j] * u.a[j][i];
            d[i] += u.b[i];
        }
        return {d[0], d[1]};
    }
} s;

std::pair<double, double> get(int i) {
    double v[2] = {double(i), 0.0};
    return s.query(i, v);
}

int ct[maxn], cl[maxn];

void rotate(int k, int v) {
    double t = double(v) / 360. * 2. * pi;
    auto uu = get(k - 1);
    SEG::node p = {{{1.0, 0.0}, {0.0, 1.0}}, {0.0, 0.0}, false};
    {
        double u[2][2] = {{1.0, 0.0}, {0.0, 1.0}};
        double w[2] = {-uu.first, -uu.second};
        p.tag(u, w);
    }
    {
        double u[2][2] = {{cos(t), sin(t)}, {-sin(t), cos(t)}};
        double w[2] = {0.0, 0.0};
        p.tag(u, w);
    }
    {
        double u[2][2] = {{1.0, 0.0}, {0.0, 1.0}};
        double w[2] = {+uu.first, +uu.second};
        p.tag(u, w);
    }
    s.modify(k, lo, p.a, p.b);
}

void extend(int k, int v) {
    auto uu = get(k - 1);
    auto vv = get(k);
    double dx = vv.first - uu.first;
    double dy = vv.second - uu.second;
    double uvl = sqrt(dx * dx + dy * dy);
    {
        double u[2][2] = {{1.0, 0.0}, {0.0, 1.0}};
        double w[2] = {dx / uvl * v, dy / uvl * v};
        s.modify(k, lo, u, w);
    }
}

int main() {
    s.init();
    int n, q;
    scanf("%d%d", &n, &q);
    for(int i = 1; i <= n; ++i) {
        ct[i] = 0;
        cl[i] = 1;
    }
    for(int i = 0; i < q; ++i) {
        int op, k, v;
        scanf("%d%d", &op, &k);
        if(op == 0) {
            scanf("%d", &v);
            rotate(k, v - ct[k]);
            ct[k] = v;
        } else if(op == 1) {
            scanf("%d", &v);
            extend(k, v - cl[k]);
            cl[k] = v;
        } else {
            auto p = get(k);
            printf("%.6f %.6f\n", p.first, p.second);
        }
    }
    return 0;
}
0