QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#202081#1831. BruteforceisaunoyaWA 8ms47412kbC++233.5kb2023-10-05 19:08:112023-10-05 19:08:12

Judging History

你现在查看的是最新测评结果

  • [2023-10-05 19:08:12]
  • 评测
  • 测评结果:WA
  • 用时:8ms
  • 内存:47412kb
  • [2023-10-05 19:08:11]
  • 提交

answer

#include <bits/stdc++.h>
#ifndef LOCAL
#define debug(...) 42
#else
#define debug(...) cerr << "[" << #__VA_ARGS__ << "]:", debug_out(__VA_ARGS__)
#endif
#define rep1(a) for (int i = 0; i < a; i++)
#define rep2(i, a) for (int i = 0; i < a; i++)
#define rep3(i, a, b) for (int i = a; i < b; i++)
#define rep4(i, a, b, c) for (int i = a; i < b; i += c)
#define overload4(a, b, c, d, e, ...) e
#define rep(...) overload4(__VA_ARGS__, rep4, rep3, rep2, rep1)(__VA_ARGS__)

#define pb emplace_back
using namespace std;
template <typename T, typename T2> void cmin(T &x, const T2 &y) {
  x = x < y ? x : y;
}
template <typename T, typename T2> void cmax(T &x, const T2 &y) {
  x = x > y ? x : y;
}
using ll = long long;
using vi = vector<int>;
using pii = pair<int, int>;
template <class T> using vc = vector<T>;
template <class T> using pq = priority_queue<T>;
template <class T> using pqg = priority_queue<T, vector<T>, greater<T>>;
mt19937 rng(time(NULL));
const int inf = 1000000000;
const ll lnf = 1000000000000000000;
#define sz(x) int((x).size())
#define all(x) begin(x), end(x)
#define fi first
#define se second

int n, k, w, q;
const int N = 131072;
const int P = 998244353;
int cnt[N * 2], s[N * 2][6], s2[N * 2][6][6];
int c[6][6], pw[N][6];

int power(int x, int y) {
  int res = 1;
  while (y) {
    if (y % 2 == 1) {
      res = 1ll * res * x % P;
    }
    x = 1ll * x * x % P;
    y /= 2;
  }
  return res;
}

int inverse(int x) { return power(x, P - 2); }

void md(int &x) {
  if (x >= P) {
    x -= P;
  }
}

void pu(int x) {
  int sz = cnt[x * 2];
  cnt[x] = cnt[x * 2] + cnt[x * 2 + 1];
  memcpy(s[x], s[x * 2], sizeof s[x]);
  for (int i = 0; i <= k; i++) {
    for (int j = 0; j <= i; j++) {
      md(s[x][i] += 1ll * s[x * 2 + 1][j] * c[i][j] % P * pw[sz][i - j] % P);
    }
  }
  memcpy(s2[x], s2[x * 2], sizeof s2[x]);
  for (int i = 0; i < w; i++) {
    for (int j = 0; j < w; j++) {
      md(s2[x][(i + sz) % w][j] += s2[x * 2 + 1][i][j]);
    }
  }
}
void upd(int x, const int &v) {
  int X = x + N;
  int &c = cnt[X];
  if (v == 1) {
    c += 1;
    for (int i = 0; i <= k; i++) {
      md(s[X][i] += 1ll * pw[c][i] * x % P);
    }
    s2[X][c % w][x % w] += 1;
  } else {
    for (int i = 0; i <= k; i++) {
      md(s[X][i] += P - 1ll * pw[c][i] * x % P);
    }
    s2[X][c % w][x % w] -= 1;
    c -= 1;
  }
  while (X >>= 1) {
    pu(X);
  }
}

void solve() {
  cin >> n >> k >> w;
  vi a(n);
  for (int i = 0; i < n; i++) {
    cin >> a[i];
    upd(a[i], 1);
  }
  cin >> q;
  const int ivw = inverse(w);
  while (q--) {
    int x, y;
    cin >> x >> y;
    --x;
    upd(a[x], -1);
    upd(a[x] = y, 1);
    int ans = s[1][k];
    for (int i = 1; i < w; i++) {
      for (int j = 1; j < w; j++) {
        int c = s2[1][i][j];
        md(ans += P - 1ll * c * (pw[i][k] % w) * j % P);
      }
    }
    cout << 1ll * ans * ivw % P << "\n";
  }
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  for (int i = 0; i < N; i++) {
    pw[i][0] = 1;
    for (int j = 1; j < 6; j++) {
      pw[i][j] = 1ll * pw[i][j - 1] * i % P;
    }
  }
  for (int i = 0; i < 6; i++) {
    c[i][0] = c[i][i] = 1;
    for (int j = 1; j < i; j++) {
      c[i][j] = c[i - 1][j - 1] + c[i - 1][j];
    }
  }
  // for (int i = 0; i < 6; i++) {
  //   for (int j = 0; j < 6; j++) {
  //     cerr << c[i][j] << " \n"[j + 1 == 6];
  //   }
  // }
  int t = 1;
  // cin >> t;
  while (t--) {
    solve();
  }
  return 0;
}

詳細信息

Test #1:

score: 100
Accepted
time: 0ms
memory: 21464kb

input:

3 1 1
2 2 8
2
2 5
3 6

output:

36
30

result:

ok 2 number(s): "36 30"

Test #2:

score: 0
Accepted
time: 0ms
memory: 21440kb

input:

4 2 2
1 3 3 7
4
1 1
2 4
3 8
4 8

output:

75
80
103
108

result:

ok 4 number(s): "75 80 103 108"

Test #3:

score: 0
Accepted
time: 0ms
memory: 39816kb

input:

10 1 1
16251 28898 58179 69362 48663 81443 34949 87167 16552 58931
10
6 89124
8 27159
4 7332
1 15852
9 67405
7 19413
10 97472
7 31114
6 31847
5 43794

output:

3511390
3107346
2780002
2779204
3079414
3018965
3365708
3406982
2970195
2936112

result:

ok 10 numbers

Test #4:

score: 0
Accepted
time: 0ms
memory: 47048kb

input:

100 2 2
44625 87890 57662 73552 89172 64466 22834 24089 60132 5187 88984 19022 67559 53954 42114 19018 80035 3367 50518 15479 72359 15452 38886 5945 34974 86214 16805 71388 48981 45377 34170 61384 88881 29453 94738 94669 56746 80508 79155 94505 82745 38134 41769 2032 23186 5636 39727 54400 86133 497...

output:

81216962
152846115
156547587
163221145
293598979
178882623
92185541
202208317
181562234
200670345
213033267
262881364
247600647
301317991
271334928
261885869
261690216
247578015
236998290
291971331
293746018
424418987
402413699
300515771
300819876
344295103
391424353
392633865
361623113
355154190
47...

result:

ok 100 numbers

Test #5:

score: -100
Wrong Answer
time: 8ms
memory: 47412kb

input:

1000 5 5
67444 21858 17070 50937 22108 62205 2999 96284 84111 16255 69173 11611 84406 28349 95817 86160 87289 19642 22706 44359 31899 36187 15946 86429 23120 65928 81187 32204 37790 18497 52182 23455 59579 78480 45277 57706 60123 99315 19014 72404 35420 14632 12210 38628 1729 18606 23941 96652 80784...

output:

448982482
318631492
90367840
811603017
536477176
692556745
62990217
201292755
656271606
39299726
904902004
682329753
415436692
172036477
307435301
263727740
240392054
817310188
279181337
609018663
744046166
313109572
146348690
684606018
105662634
927540135
395442092
940075695
928045058
210861090
871...

result:

wrong answer 1st numbers differ - expected: '448982964', found: '448982482'