結果

問題 No.2229 Treasure Searching Rod (Hard)
ユーザー tnakao0123
提出日時 2023-04-24 11:22:59
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 62 ms / 2,000 ms
コード長 915 bytes
コンパイル時間 464 ms
コンパイル使用メモリ 41,924 KB
実行使用メモリ 5,504 KB
最終ジャッジ日時 2024-11-08 13:53:08
合計ジャッジ時間 4,621 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

/* -*- coding: utf-8 -*-
 *
 * 2229.cc:  No.2229 Treasure Searching Rod (Hard) - yukicoder
 */

#include<cstdio>
#include<algorithm>
 
using namespace std;

/* constant */

const int MAX_K = 200000;
const int MOD = 998244353;

/* typedef */

typedef long long ll;

/* global variables */

int xs[MAX_K], ys[MAX_K], vs[MAX_K];

/* subroutines */

inline ll sigma(int n) { return (ll)n * (n + 1) / 2; }

/* main */

int main() {
  int h, w, k;
  scanf("%d%d%d", &h, &w, &k);
  for (int i = 0; i < k; i++)
    scanf("%d%d%d", xs + i, ys + i, vs + i);

  int sum = 0;
  for (int i = 0; i < k; i++) {
    int dx = xs[i];
    int y0 = max(0, dx - ys[i]), y1 = max(0, ys[i] + dx - 1 - w);
    ll s0 = sigma(dx) - sigma(y0), s1 = sigma(dx) - sigma(y1);
    ll s = s0 + s1 - dx;
    //printf("x=%d,y=%d -> s=%lld\n", xs[i], ys[i], s);
    sum = (sum + s % MOD * vs[i] % MOD) % MOD;
  }

  printf("%d\n", sum);
  return 0;
}
0