QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#132774#5464. Dice GameClHg2WA 9ms3640kbC++142.6kb2023-07-31 15:44:472023-07-31 15:44:48

Judging History

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

  • [2023-08-10 23:21:45]
  • System Update: QOJ starts to keep a history of the judgings of all the submissions.
  • [2023-07-31 15:44:48]
  • 评测
  • 测评结果:WA
  • 用时:9ms
  • 内存:3640kb
  • [2023-07-31 15:44:47]
  • 提交

answer

#include <array>
#include <cstddef>
#include <cstdint>
#include <fstream>
#include <iostream>
#include <string>

namespace {
using std::cin;
using std::cout;
using std::int64_t;
using std::size_t;

namespace base {
template <typename T, size_t... sizes>
struct NestedArray {};

template <typename T, size_t size, size_t... sizes>
struct NestedArray<T, size, sizes...> {
  using Type = std::array<typename NestedArray<T, sizes...>::Type, size>;
};

template <typename T>
struct NestedArray<T> {
  using Type = T;
};

template <typename T, size_t... sizes>
using Array = typename NestedArray<T, sizes...>::Type;

void OptimizeIO() {
  std::ios::sync_with_stdio(false);
  cin.tie(nullptr), cout.tie(nullptr);
}

void OptimizeIO(const std::string &input_file, const std::string &output_file) {
  static std::ifstream input_stream(input_file);
  static std::ofstream output_stream(output_file);
  cin.rdbuf(input_stream.rdbuf()), cout.rdbuf(output_stream.rdbuf());
  cin.tie(nullptr), cout.tie(nullptr);
}
}  // namespace base

using base::Array;

const int kP = 998244353, kInv2 = (kP + 1) / 2;

namespace mod {
inline void Mul(int &a, int b) { a = int64_t{a} * b % kP; }

int Pow(int a, int b) {
  int ans = 1, prod = a;

  while (b) {
    if (b & 1) Mul(ans, prod);
    Mul(prod, prod), b >>= 1;
  }

  return ans;
}

inline int Inv(int a) { return Pow(a, kP - 2); }
}  // namespace mod

const int kMaxLogN = 30;
int n;
int64_t ans;
Array<int64_t, kMaxLogN> c, sum;

inline int Calc(int n, int k) {
  int q = (n - 1) >> (k + 1), r = n - (q << (k + 1));
  return (q << k) + std::max(0, r - (1 << k));
}

void Dfs(int cur, bool f, int64_t sum) {
  if (cur == -1) {
    ans += std::max(sum % kP, int64_t{0});
    return;
  }

  if (sum + ::sum[cur] <= 0) return;

  if (sum - ::sum[cur] >= 0 && !f) {
    ans += sum << (cur + 1);
    return;
  }

  if (f) {
    if ((n - 1) >> cur & 1) {
      Dfs(cur - 1, true, sum - c[cur]);
      Dfs(cur - 1, false, sum + c[cur]);
    } else {
      Dfs(cur - 1, true, sum + c[cur]);
    }
  } else {
    Dfs(cur - 1, false, sum - c[cur]);
    Dfs(cur - 1, false, sum + c[cur]);
  }
}

void Solve() {
  cin >> n;

  for (int i = 0; i < kMaxLogN; ++i) {
    sum[i] = c[i] = int64_t{Calc(n, i)} << i;
    if (i) sum[i] += sum[i - 1];
  }

  ans = 0, Dfs(kMaxLogN - 1, true, 0);
  int inv = mod::Inv(n);
  cout << (ans % kP * inv % kP * inv + int64_t{n - 1} * kInv2) % kP << "\n";
}

int Main() {
  base::OptimizeIO();
  int t;
  cin >> t;
  while (t--) Solve();
  return 0;
}
}  // namespace

int main() { return Main(); }

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 1ms
memory: 3460kb

input:

4
1
2
3
4

output:

0
249561089
776412276
2

result:

ok 4 number(s): "0 249561089 776412276 2"

Test #2:

score: 0
Accepted
time: 1ms
memory: 3384kb

input:

100
119
75
29
10
17
29
449
71
72
12
79
117
83
80
35
272
105
497
346
287
362
140
297
167
111
419
210
212
170
413
373
210
196
39
1
101
258
496
333
293
392
2
187
431
157
342
436
106
449
136
378
243
357
325
237
254
22
292
62
435
18
446
471
18
42
377
181
350
19
389
212
58
45
70
52
63
107
71
66
355
381
30...

output:

645006489
296012775
400009943
299473312
221064504
400009943
60548260
896063466
573066250
471393174
175624519
531171954
143020402
134763040
560646647
43176836
269095960
284396636
191400715
895243672
967774080
745220060
584864173
5941827
724073586
701650152
262576881
417830609
833275086
916357319
1435...

result:

ok 100 numbers

Test #3:

score: 0
Accepted
time: 1ms
memory: 3424kb

input:

10
321
4212
2977
3438
679
2518
3359
2348
861
1853

output:

115992677
210903174
216412050
780704799
956649209
200804004
267658252
194263892
880330717
328032438

result:

ok 10 numbers

Test #4:

score: 0
Accepted
time: 1ms
memory: 3640kb

input:

12
3324
3167
3780
2887
1541
494
4374
3416
4036
3637
3888
4630

output:

458228110
29999980
333283438
544178470
753289945
257353919
515561689
628492255
799560584
924311157
17047279
109086825

result:

ok 12 numbers

Test #5:

score: -100
Wrong Answer
time: 9ms
memory: 3432kb

input:

10000
11424730
231352165
835903988
101946284
358678826
610737477
100503274
909725926
510714513
407317433
668332866
442951503
582321912
117772499
180141789
570384258
375362779
724696666
959258227
178470161
245672142
629464136
515375633
58812672
119047157
272587596
993112981
678095944
67996298
2717673...

output:

-414914768
-960869639
634066993
-250588676
-991055996
825789992
158597923
115444556
-690890751
290672888
664835621
811427905
533022457
823378799
300963738
259403998
593644796
-99672104
493303907
-189472973
699154294
551985033
904515543
-758470944
436162921
494492992
873446731
458378729
-290334761
79...

result:

wrong answer 1st numbers differ - expected: '773455292', found: '-414914768'