結果

問題 No.2229 Treasure Searching Rod (Hard)
ユーザー boatmusclesboatmuscles
提出日時 2023-02-24 21:58:40
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 62 ms / 2,000 ms
コード長 1,557 bytes
コンパイル時間 4,313 ms
コンパイル使用メモリ 260,860 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-09-13 05:27:58
合計ジャッジ時間 6,747 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/extc++.h>
#include<atcoder/modint>
using namespace __gnu_pbds;
using namespace std;
using namespace atcoder;
using ll = long long;
using ull = unsigned long long;
using Mint = modint998244353;
using mint = modint;
#define rep(i, n) for (int i = 0; i < (int)(n); ++i)
#define rrep(i, n) for (int i = (int)(n)-1; i >= 0; --i)
#define rep2(i, a, b) for (int i = (int)a; i < (int)(b); ++i)
#define rrep2(i, a, b) for (int i = (int)(b)-1; i >= (int)(a); --i)
template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }
constexpr int dx[] = {-1,0,1,0};
constexpr int dy[] = {0,-1,0,1};
constexpr int MAX_N = 100000;
//ダイクストラ法：makedijkstra
//エラトステネスの篩：makesieve
//テストケースが複数の場合：multest

int main(){
    //cin.tie(nullptr);
    //std::ios_base::sync_with_stdio(false);
    int H, W, total_testcases;
    scanf("%d %d %d", &H, &W, &total_testcases);
    auto add = [&](int x, int y){
        bool right = x + y > W + 1;
        bool left = x > y;
        return (right ? (ll)x*(W - y + 1) - (ll)(W - y + 1)*(W - y)/2 : (ll)x*(x+1)/2)
            + (left ? (ll)x*y - (ll)y*(y-1)/2 : (ll)x*(x+1)/2)
            - x;
    };
    Mint ans{0};
    for (int testcase_i = 1; testcase_i <= total_testcases; ++testcase_i){
        int x, y, v;
        scanf("%d %d %d", &x, &y, &v);
        ans += Mint(add(x, y))*Mint(v);
    }
    printf("%u\n", ans.val());
    return 0;
}
0