結果

問題 No.1680 Sum and Difference
ユーザー sten_sansten_san
提出日時 2021-09-17 22:34:48
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2 ms / 1,000 ms
コード長 1,815 bytes
コンパイル時間 1,717 ms
コンパイル使用メモリ 203,540 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-29 21:18:58
合計ジャッジ時間 2,371 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 1 ms
6,944 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 1 ms
6,940 KB
testcase_07 AC 1 ms
6,940 KB
testcase_08 AC 1 ms
6,944 KB
testcase_09 AC 1 ms
6,944 KB
testcase_10 AC 1 ms
6,940 KB
testcase_11 AC 2 ms
6,940 KB
testcase_12 AC 2 ms
6,940 KB
testcase_13 AC 2 ms
6,944 KB
testcase_14 AC 1 ms
6,940 KB
testcase_15 AC 2 ms
6,940 KB
testcase_16 AC 2 ms
6,944 KB
testcase_17 AC 1 ms
6,940 KB
testcase_18 AC 2 ms
6,940 KB
testcase_19 AC 1 ms
6,940 KB
testcase_20 AC 2 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

struct iofast_t {
    iofast_t() {
        ios::sync_with_stdio(false);
        cin.tie(nullptr);
    }
} iofast;

struct uns_t {} uns;
template <typename Element, typename Head, typename ...Args>
auto vec(Element init, Head arg, Args ...args) {
    if constexpr (sizeof...(Args) == 0) return std::vector(arg, init);
    else return std::vector(arg, vec(init, args...));
}
template <typename Element, typename Head, typename ...Args>
auto vec(uns_t, Head arg, Args ...args) {
    return vec(Element(), arg, args...);
}

template <typename T, typename Compare = less<T>>
T &chmin(T &l, T r, Compare &&f = less<T>()) { return l = min(l, r, f); }
template <typename T, typename Compare = less<T>>
T &chmax(T &l, T r, Compare &&f = less<T>()) { return l = max(l, r, f); }

#include <atcoder/modint>
using mint = atcoder::modint1000000007;

int64_t sign(int64_t x) {
    return clamp<int64_t>(x, -1, 1);
}

int64_t floor(int64_t x, int64_t y) {
    if (0 < sign(x) * sign(y)) {
        return x / y;
    }
    return -((abs(x) + abs(y) - 1) / abs(y));
}

int main() {
    int64_t a, b; cin >> a >> b;

    auto ab = floor(a - b, 2);
    auto ba = floor(b - a, 2);

    mint ans = 0;
    if (ab < ba) {
        ans += mint(ba - ab) * (2 * a + 1);
    }
    if (ba < ab) {
        ans += mint(ab - ba) * (2 * b + 1);
    }

    if (floor(-(a + b + 1), 2) < min(ab, ba)) {
        auto l = floor(-(a + b + 1), 2);
        auto r = min(ab, ba);
        ans += mint(a + b + 1) * (r - l) + mint(r - l) * (2 * (l + 1) + (r - l - 1));
    }
    if (max(ab, ba) < floor(a + b + 1, 2)) {
        auto l = max(ab, ba);
        auto r = floor(a + b + 1, 2);
        ans += mint(a + b + 1) * (r - l) - mint(r - l) * (2 * (l + 1) + (r - l - 1));
    }

    cout << ans.val() << endl;
}
0