#551680 (C++17) No.1226 I hate Robot Arms

提出ソース

結果

問題	No.1226 I hate Robot Arms
ユーザー	nok0
提出日時	2020-09-11 22:34:15
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	1,581 ms / 2,000 ms
コード長	10,242 bytes
コンパイル時間	2,020 ms
コンパイル使用メモリ	213,688 KB
実行使用メモリ	16,928 KB
最終ジャッジ日時	2024-06-10 10:26:57
合計ジャッジ時間	34,308 ms
ジャッジサーバーID （参考情報）	judge5 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,816 KB
testcase_01	AC	4 ms 6,940 KB
testcase_02	AC	453 ms 6,940 KB
testcase_03	AC	533 ms 7,040 KB
testcase_04	AC	731 ms 16,452 KB
testcase_05	AC	721 ms 16,780 KB
testcase_06	AC	1,467 ms 16,684 KB
testcase_07	AC	404 ms 6,940 KB
testcase_08	AC	336 ms 16,640 KB
testcase_09	AC	980 ms 7,032 KB
testcase_10	AC	110 ms 6,940 KB
testcase_11	AC	775 ms 16,460 KB
testcase_12	AC	407 ms 10,148 KB
testcase_13	AC	487 ms 16,824 KB
testcase_14	AC	1,131 ms 7,156 KB
testcase_15	AC	329 ms 16,688 KB
testcase_16	AC	1,435 ms 16,648 KB
testcase_17	AC	382 ms 6,944 KB
testcase_18	AC	281 ms 6,940 KB
testcase_19	AC	806 ms 16,624 KB
testcase_20	AC	725 ms 16,452 KB
testcase_21	AC	1,184 ms 16,656 KB
testcase_22	AC	1,506 ms 16,760 KB
testcase_23	AC	1,479 ms 16,812 KB
testcase_24	AC	1,504 ms 16,876 KB
testcase_25	AC	1,492 ms 16,848 KB
testcase_26	AC	1,483 ms 16,808 KB
testcase_27	AC	1,542 ms 16,928 KB
testcase_28	AC	1,581 ms 16,648 KB
testcase_29	AC	1,572 ms 16,764 KB

権限があれば一括ダウンロードができます

コンパイルメッセージ

main.cpp: In function 'int main()':
main.cpp:292:32: warning: narrowing conversion of 'c' from 'long double' to 'int' [-Wnarrowing]
  292 |             ST.set(i, {{x, c}, c});
      |                                ^

ソースコード

raw source code

/**
 *	author: nok0
 *	created: 2020.09.11 21:29:14
**/
#ifdef LOCAL
#define _GLIBCXX_DEBUG
#endif
#include<bits/stdc++.h>
using namespace std;
#pragma region Macros
#define ll long long
#define ld long double
#define FOR(i,l,r) for(int i=(l);i<(r);++i)
#define REP(i,n) FOR(i,0,n)
#define REPS(i,n) FOR(i,1,n+1)
#define RFOR(i,l,r) for(int i=(l);i>=(r);--i)
#define RREP(i,n) RFOR(i,n-1,0)
#define RREPS(i,n) RFOR(i,n,1)
#define pb push_back
#define eb emplace_back
#define SZ(x) ((int)(x).size())
#define all(x) (x).begin(),(x).end()
#define rall(x) (x).rbegin(),(x).rend()
template<class T = int> using V = vector<T>;
template<class T = int> using VV = V<V<T>>;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
#define VEC(type, name, size)\
    V<type> name(size);\
    IN(name)
#define VVEC(type, name, h, w)\
    VV<type> name(h, V<type>(w));\
    IN(name)
#define INT(...)\
    int __VA_ARGS__;\
    IN(__VA_ARGS__)
#define LL(...)\
    ll __VA_ARGS__;\
    IN(__VA_ARGS__)
#define STR(...)\
    string __VA_ARGS__;\
    IN(__VA_ARGS__)
#define CHAR(...)\
    char __VA_ARGS__;\
    IN(__VA_ARGS__)
#define DOUBLE(...)\
    DOUBLE __VA_ARGS__;\
    IN(__VA_ARGS__)
#define LD(...)\
    LD __VA_ARGS__;\
    IN(__VA_ARGS__)
template <class T> void scan(T a) { cin >> a; }
void scan(int &a) { cin >> a; }
void scan(long long &a) { cin >> a; }
void scan(char &a) { cin >> a; }
void scan(double &a) { cin >> a; }
void scan(long double &a) { cin >> a; }
void scan(char a[]) { scanf("%s", a); }
void scan(string &a) { cin >> a; }
template <class T> void scan(V<T> &);
template <class T, class L> void scan(pair<T, L> &);
template <class T> void scan(V<T> &a) { for(auto &i : a) scan(i); }
template <class T, class L> void scan(pair<T, L> &p){ scan(p.first); scan(p.second); }
template <class T> void scan(T &a) { cin >> a; }
void IN() {}
template <class Head, class... Tail> void IN(Head &head, Tail &... tail) { scan(head); IN(tail...); }
template <class T> inline void print(T x){ cout << x << '\n';}
template <typename T1, typename T2>
istream &operator>>(istream &is, pair<T1, T2> &p){
    is >> p.first >> p.second;
    return is;
}
template <typename T1, typename T2>
ostream &operator<<(ostream &os, const pair<T1, T2> &p){
    os << p.first << " " << p.second;
    return os;
}
template <class T>
ostream &operator<<(ostream &os, const V<T> &v){
    REP(i, SZ(v)){
        if(i) os << " ";
        os << v[i];
    }
    return os;
}
//debug
template <typename T>
void view(const V<T> &v){
    cerr << "{ ";
    for(const auto &e : v){
        cerr << e << ", ";
    }
    cerr << "\b\b }";
}
template <typename T>
void view(const VV<T> &vv){
    cerr << "{\n";
    for(const auto &v : vv){
        cerr << "\t";
        view(v);
        cerr << "\n";
    }
    cerr << "}";
}
template <typename T, typename U>
void view(const V<pair<T, U>> &v){
    cerr << "{\n";
    for(const auto &c : v) cerr << "\t(" << c.first << ", " << c.second << ")\n";
    cerr << "}";
}
template <typename T, typename U>
void view(const map<T, U> &m){
    cerr << "{\n";
    for(auto &t : m) cerr << "\t[" << t.first << "] : " << t.second << "\n";
    cerr << "}";
}
template <typename T, typename U>
void view(const pair<T, U> &p){ cerr << "(" << p.first << ", " << p.second << ")";}
template <typename T>
void view(const set<T> &s) {
    cerr << "{ ";
    for(auto &t : s) {
        view(t);
        cerr << ", ";
    }
    cerr << "\b\b }";
}
template <typename T>
void view(T e) { cerr << e;}
#ifdef LOCAL
void debug_out() {}
template <typename Head, typename... Tail>
void debug_out(Head H, Tail... T) {
    view(H);
    cerr << ", ";
    debug_out(T...);
}
#define debug(...)                                           \
    do{                                                     \
        cerr << __LINE__ << " [" << #__VA_ARGS__ << "] : ["; \
        debug_out(__VA_ARGS__);                              \
        cerr << "\b\b]\n";                                   \
    } while(0)
#else
#define debug(...) (void(0))
#endif
template <class T> V<T> press(V<T> &x){ 
    V<T> res = x;
    sort(all(res));
    res.erase(unique(all(res)), res.end());
    REP(i, SZ(x)){
        x[i] = lower_bound(all(res), x[i]) - res.begin();
    }
    return res;
}
template<class T> inline bool chmin(T& a, T b) {if (a > b) {a = b; return true; }return false; }
template<class T> inline bool chmax(T& a, T b) {if (a < b) {a = b; return true; }return false; }
inline void Yes(bool b = true) {cout << (b ? "Yes" : "No") << '\n';}
inline void YES(bool b = true) {cout << (b ? "YES" : "NO") << '\n';}
inline void err(bool b = true) {if(b) {cout << -1 << '\n'; exit(0);}}
template<class T> inline void fin(bool b = true, T e = 0) {if(b) {cout << e << '\n'; exit(0);}}
template<class T> T divup(T x, T y) {return (x+(y-1))/y;}
template <typename T> T pow(T a, long long n, T e = 1) {T ret = e; while (n) {if (n & 1) ret *= a; a *= a; n >>= 1; } return ret; }
struct iofast{iofast(){ios::sync_with_stdio(false); cin.tie(nullptr); cout << fixed << setprecision(15);}}iofast_;
const int inf = 1e9;
const ll INF = 1e18;
#pragma endregion

//Segment Tree
//reference materials: <https://www.creativ.xyz/segment-tree-abstraction-979/>, <https://algo-logic.info/segment-tree/>
template <class Monoid>
class SegTree {
    using F = function<Monoid(Monoid,Monoid)>;
    int n;// 葉の数
    vector<Monoid> data; // データを格納する配列
    Monoid def;          // 初期値かつ単位元
    F operation; // 区間クエリ関数
    F update;    // 点更新関数
    // 区間[a,b)の総和。ノードk=[l,r)に着目
    Monoid _query(int a, int b, int k, int l, int r) {
        if (r <= a || b <= l) return def; // 交差しない
        if (a <= l && r <= b)
            return data[k]; // a,l,r,bの順で完全に含まれる
        else {
            Monoid c1 = _query(a, b, 2 * k + 1, l, (l + r) / 2); // 左の子
            Monoid c2 = _query(a, b, 2 * k + 2, (l + r) / 2, r); // 右の子
            return operation(c1, c2);
        }
    }
 
  public:
    // _n:SegTreeのサイズ, _def:初期値かつ単位元, _operation:クエリ関数,
    // _update:点更新関数
    SegTree(size_t _n, Monoid _def, F _operation, F _update)
        : def(_def), operation(_operation), update(_update) {
        n = 1;
        while (n < _n) {
            n *= 2;
        }
        data = vector<Monoid>(2 * n - 1, def);
    }
 
    // 場所i(0-indexed)の値をxで更新
    void set(int i, Monoid x) {
        i += n - 1;
        data[i] = update(data[i], x);
        while (i > 0) {
            i = (i - 1) / 2;
            data[i] = operation(data[i * 2 + 1], data[i * 2 + 2]);
        }
    }
 
    // 半開区間[a, b)の区間クエリ
    Monoid query(int a, int b) {
        return _query(a, b, 0, 0, n);
    }
 
    // 添字アクセス
    Monoid operator[](int i) {
        return data[i + n - 1];
    }
    // 半開区間[a,b)でx以下の要素を持つ最右位置を返す(二分探索)
    // a:半開区間の左端, b:半開区間の右端, x:x以下の要素を求める
    int find_rightest(int a, int b, Monoid x) { return find_rightest_sub(a, b, x, 0, 0, n); }
    // 半開区間[a,b)でx以下の要素を持つ最左位置を返す(二分探索)
    // a:半開区間の左端, b:半開区間の右端, x:x以下の要素を求める
    int find_leftest(int a, int b, Monoid x) { return find_leftest_sub(a, b, x, 0, 0, n); }
    int find_rightest_sub(int a, int b, Monoid x, int k, int l, int r) {
        if (data[k] > x || r <= a || b <= l) {  // 自分の値がxより大きい or [a,b)が[l,r)の範囲外ならreturn a-1
            return a - 1;
        } else if (k >= n - 1) {  // 自分が葉ならその位置をreturn
            return (k - (n - 1));
        } else {
            int vr = find_rightest_sub(a, b, x, 2 * k + 2, (l + r) / 2, r);
            if (vr != a - 1) {  // 右の部分木を見て a-1 以外ならreturn
                return vr;
            } else {  // 左の部分木を見て値をreturn
                return find_rightest_sub(a, b, x, 2 * k + 1, l, (l + r) / 2);
            }
        }
    }
    int find_leftest_sub(int a, int b, Monoid x, int k, int l, int r) {
        if (data[k] > x || r <= a || b <= l) {  // 自分の値がxより大きい or [a,b)が[l,r)の範囲外ならreturn b
            return b;
        } else if (k >= n - 1) {  // 自分が葉ならその位置をreturn
            return (k - (n - 1));
        } else {
            int vl = find_leftest_sub(a, b, x, 2 * k + 1, l, (l + r) / 2);
            if (vl != b) {  // 左の部分木を見て b 以外ならreturn
                return vl;
            } else {  // 右の部分木を見て値をreturn
                return find_leftest_sub(a, b, x, 2 * k + 2, (l + r) / 2, r);
            }
        }
    }
};

const ld pi = acos(-1.0);
ld ch(ld a){return pi * a / 180.0;};
ld rch(ld a){return a * 180.0 / pi;};

int main(){
    INT(n, q);
    using P = pair<pair<ld, ld>, int>;
    using cmp = complex<ld>;
    auto ope = [](P a, P b){
        ld px = b.first.first * cos(ch(b.first.second));
        ld py = b.first.first * sin(ch(b.first.second));
        cmp u(px, py);
        u *= cmp(cos(ch(a.second)), sin(ch(a.second)));
        ld x = a.first.first * cos(ch(a.first.second)) + u.real();
        ld y = a.first.first * sin(ch(a.first.second)) + u.imag();
        cmp z(x, y);
        return P({abs(z), rch(arg(z))}, (a.second + b.second) % 360);
    };
    auto upd = [](P a, P b){return b;};
    SegTree<P> ST(n, {{0, 0}, 0}, ope, upd);
    REP(i, n) ST.set(i, {{1, 0}, 0});
    V<pii> data(n, {1, 0});
    while(q--){
        INT(type);
        if(type == 0){
            INT(i, x); i--;
            data[i].second = x;
            ld c = data[i].first;
            ST.set(i, {{c, x}, x});
        }
        if(type == 1){
            INT(i, x); i--;
            data[i].first = x;
            ld c = data[i].second;
            ST.set(i, {{x, c}, c});
        }
        if(type == 2){
            INT(i);
            P p = ST.query(0, i);
            cout << p.first.first * cos(ch(p.first.second)) << " " << p.first.first * sin(ch(p.first.second)) << endl;
        }
    }
}

yukicoder

結果

テストケース

コンパイルメッセージ

ソースコード