結果

問題 No.3048 Swing
ユーザー SnowBeenDiding
提出日時 2025-03-07 21:39:31
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 2,074 bytes
コンパイル時間 5,876 ms
コンパイル使用メモリ 332,416 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2025-06-20 02:24:37
合計ジャッジ時間 7,011 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 62
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <atcoder/all>
#include <bits/stdc++.h>
#define rep(i, a, b) for (ll i = (ll)(a); i < (ll)(b); i++)
using namespace atcoder;
using namespace std;

typedef long long ll;

void pl(ll x, ll n) {
    auto check = [&](ll mid) { // ([...,1,1,1,0,0,0,...])
        if (mid < 0)
            return true;
        if (mid * (mid + 1) / 2 >= x)
            return true;
        return false;
    };
    auto binary = [&]() {
        ll ac = 1414213572, wa = -1;
        while (ac - wa > 1) {
            ll mid = (ac + wa) / 2;
            if (check(mid))
                ac = mid;
            else
                wa = mid;
        }
        return ac;
    };
    ll minus_ind = binary();
    if (n <= minus_ind) {
        cout << x - n * (n + 1) / 2 << endl;
        return;
    }
    ll mi = x - minus_ind * (minus_ind + 1) / 2;
    ll pl = mi + minus_ind + 1;
    ll amari = n - minus_ind;
    ll ans;
    if (amari & 1LL) {
        ans = pl + amari / 2;
    } else {
        ans = mi - amari / 2;
    }
    cout << ans << endl;
}

void mi(ll x, ll n) {
    x *= -1;
    auto check = [&](ll mid) { // ([...,1,1,1,0,0,0,...])
        if (mid < 0)
            return true;
        if (mid * (mid + 1) / 2 > x)
            return true;
        return false;
    };
    auto binary = [&]() {
        ll ac = 1414213572, wa = -1;
        while (ac - wa > 1) {
            ll mid = (ac + wa) / 2;
            if (check(mid))
                ac = mid;
            else
                wa = mid;
        }
        return ac;
    };
    ll minus_ind = binary();
    if (n <= minus_ind) {
        cout << -(x - n * (n + 1) / 2) << endl;
        return;
    }
    ll mi = x - minus_ind * (minus_ind + 1) / 2;
    ll pl = mi + minus_ind + 1;
    ll amari = n - minus_ind;
    ll ans;
    if (amari & 1LL) {
        ans = pl + amari / 2;
    } else {
        ans = mi - amari / 2;
    }
    cout << -ans << endl;
}

int main() {
    cin.tie(0);
    cout.tie(0);
    ios::sync_with_stdio(0);
    ll x, n;
    cin >> x >> n;
    if (x >= 0)
        pl(x, n);
    else
        mi(x, n);
}
0