#954781 (C) No.2653 [Cherry 6th Tune] Re: start! (Make it Zero!)

提出ソース

結果

問題	No.2653 [Cherry 6th Tune] Re: start! (Make it Zero!)
ユーザー	👑 chro_96
提出日時	2024-02-23 23:37:37
言語	C (gcc 12.3.0)
結果	AC
実行時間	124 ms / 2,000 ms
コード長	2,167 bytes
コンパイル時間	341 ms
コンパイル使用メモリ	33,408 KB
実行使用メモリ	7,364 KB
最終ジャッジ日時	2024-02-23 23:37:45
合計ジャッジ時間	8,359 ms
ジャッジサーバーID （参考情報）	judge13 / judge16

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	4 ms 7,364 KB
testcase_01	AC	58 ms 7,364 KB
testcase_02	AC	89 ms 7,364 KB
testcase_03	AC	88 ms 7,364 KB
testcase_04	AC	110 ms 7,364 KB
testcase_05	AC	107 ms 7,364 KB
testcase_06	AC	46 ms 7,364 KB
testcase_07	AC	47 ms 7,364 KB
testcase_08	AC	117 ms 7,364 KB
testcase_09	AC	124 ms 7,364 KB
testcase_10	AC	116 ms 7,364 KB
testcase_11	AC	94 ms 7,364 KB
testcase_12	AC	105 ms 7,364 KB
testcase_13	AC	46 ms 7,364 KB
testcase_14	AC	46 ms 7,364 KB
testcase_15	AC	46 ms 7,364 KB
testcase_16	AC	45 ms 7,364 KB
testcase_17	AC	46 ms 7,364 KB
testcase_18	AC	45 ms 7,364 KB
testcase_19	AC	46 ms 7,364 KB
testcase_20	AC	46 ms 7,364 KB
testcase_21	AC	46 ms 7,364 KB
testcase_22	AC	46 ms 7,364 KB
testcase_23	AC	61 ms 7,364 KB
testcase_24	AC	68 ms 7,364 KB
testcase_25	AC	63 ms 7,364 KB
testcase_26	AC	66 ms 7,364 KB
testcase_27	AC	68 ms 7,364 KB
testcase_28	AC	65 ms 7,364 KB
testcase_29	AC	66 ms 7,364 KB
testcase_30	AC	65 ms 7,364 KB
testcase_31	AC	66 ms 7,364 KB
testcase_32	AC	65 ms 7,364 KB
testcase_33	AC	56 ms 7,364 KB
testcase_34	AC	79 ms 7,364 KB
testcase_35	AC	80 ms 7,364 KB
testcase_36	AC	83 ms 7,364 KB
testcase_37	AC	83 ms 7,364 KB
testcase_38	AC	81 ms 7,364 KB
testcase_39	AC	81 ms 7,364 KB
testcase_40	AC	67 ms 7,364 KB
testcase_41	AC	94 ms 7,364 KB
testcase_42	AC	68 ms 7,364 KB
testcase_43	AC	47 ms 7,364 KB
testcase_44	AC	47 ms 7,364 KB
testcase_45	AC	48 ms 7,364 KB
testcase_46	AC	47 ms 7,364 KB
testcase_47	AC	47 ms 7,364 KB
testcase_48	AC	47 ms 7,364 KB
testcase_49	AC	47 ms 7,364 KB
testcase_50	AC	48 ms 7,364 KB
testcase_51	AC	47 ms 7,364 KB
testcase_52	AC	47 ms 7,364 KB
testcase_53	AC	89 ms 7,364 KB
testcase_54	AC	99 ms 7,364 KB
testcase_55	AC	55 ms 7,364 KB
testcase_56	AC	84 ms 7,364 KB
testcase_57	AC	93 ms 7,364 KB
testcase_58	AC	54 ms 7,364 KB
testcase_59	AC	62 ms 7,364 KB
testcase_60	AC	100 ms 7,364 KB
testcase_61	AC	110 ms 7,364 KB
testcase_62	AC	88 ms 7,364 KB
testcase_63	AC	29 ms 7,364 KB
testcase_64	AC	30 ms 7,364 KB
testcase_65	AC	29 ms 7,364 KB
testcase_66	AC	31 ms 7,364 KB
testcase_67	AC	31 ms 7,364 KB
testcase_68	AC	31 ms 7,364 KB
testcase_69	AC	31 ms 7,364 KB
testcase_70	AC	31 ms 7,364 KB
testcase_71	AC	31 ms 7,364 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <stdio.h>

long long power_mod (long long a, long long b, long long mod_num) {
  long long ans = 1LL;
  
  if (b > 0LL) {
    ans = power_mod(a, b/2LL, mod_num);
    ans = (ans * ans) % mod_num;
    if (b%2LL == 1LL) {
      ans = (ans * (a % mod_num)) % mod_num;
    }
  }
  
  return ans;
}

int main () {
  int t = 0;
  int n = 0;
  long long m = 0LL;
  int x[200000] = {};
  
  int res = 0;
  
  long long mod_num = 998244353LL;
  
  long long pow[200001] = {};
  int cnt[200001][2] = {};
  long long sum[200001][2] = {};
  
  res = scanf("%d", &t);
  while (t > 0) {
    long long ans = 0LL;
    res = scanf("%d", &n);
    res = scanf("%lld", &m);
    for (int i = 0; i < n; i++) {
      res = scanf("%d", x+i);
    }
    for (int i = 0; i < n; i++) {
      cnt[i+1][0] = cnt[i][0];
      sum[i+1][0] = sum[i][0];
      if (x[i] < 0) {
        cnt[i+1][0]++;
      } else {
        sum[i+1][0] += (long long)(x[i]);
      }
    }
    if (sum[n][0]%m != 0LL && cnt[n][0] <= 0) {
      printf("0\n");
    } else if (cnt[n][0] <= 0) {
      ans = (long long)(n-1);
      for (int i = 0; i < n-1; i++) {
        if (sum[i+1][0]%m == 0LL) {
          ans -= 1LL;
        }
      }
      printf("%lld\n", ans%mod_num);
    } else {
      ans = power_mod(m, (long long)(cnt[n][0]-1), mod_num);
      ans *= (long long)(n-1);
      ans %= mod_num;
      cnt[n][1] = 0;
      sum[n][1] = 0LL;
      for (int i = n-1; i >= 0; i--) {
        cnt[i][1] = cnt[i+1][1];
        sum[i][1] = sum[i+1][1];
        if (x[i] < 0) {
          cnt[i][1]++;
        } else {
          sum[i][1] += (long long)x[i];
        }
      }
      for (int i = 0; i < n-1; i++) {
        if (cnt[i+1][0] <= 0 && sum[i+1][0]%m == 0LL) {
          ans += mod_num-power_mod(m, (long long)(cnt[i][1]-1), mod_num);
        } else if (cnt[i+1][1] <= 0 && sum[i+1][1]%m == 0LL) {
          ans += mod_num-power_mod(m, (long long)(cnt[i+1][0]-1), mod_num);
        } else if (cnt[i+1][0] > 0 && cnt[i+1][1] > 0) {
          ans += mod_num-power_mod(m, (long long)(cnt[i+1][0]+cnt[i+1][1]-2), mod_num);
        }
      }
      printf("%lld\n", ans%mod_num);
    }
    t--;
  }
  
  return 0;
}

yukicoder

結果

テストケース

ソースコード