結果

問題 No.2711 Connecting Lights
ユーザー tnakao0123
提出日時 2024-04-30 10:29:21
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 39 ms / 5,000 ms
コード長 1,053 bytes
コンパイル時間 314 ms
コンパイル使用メモリ 41,856 KB
最終ジャッジ日時 2025-02-21 09:52:08
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:32:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   32 |   scanf("%d%d%d", &n, &m, &k);
      |   ~~~~~^~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff #

/* -*- coding: utf-8 -*-
 *
 * 2711.cc:  No.2711 Connecting Lights - yukicoder
 */

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

/* constant */

const int MAX_N = 5;
const int NBITS = 1 << MAX_N;
const int MAX_M = 20000;
const int MOD = 998244353;

/* typedef */

/* global variables */

int bnums[NBITS], dp[MAX_M][NBITS];

/* subroutines */

inline void addmod(int &a, int b) { a = (a + b) % MOD; }

/* main */

int main() {
  int n, m, k;
  scanf("%d%d%d", &n, &m, &k);

  int nbits = 1 << n;
  for (int bits = 1, msb = 1; bits < nbits; bits++) {
    if ((msb << 1) <= bits) msb <<= 1;
    bnums[bits] = bnums[bits ^ msb] + 1;
  }

  fill(dp[0], dp[0] + nbits, 1);
  for (int i = 0; i + 1 < m; i++)
    for (int bits0 = 0; bits0 < nbits; bits0++)
      if ((int)dp[i][bits0])
	for (int bits1 = 0; bits1 < nbits; bits1++)
	  if (bnums[bits0 & bits1] >= k)
	    addmod(dp[i + 1][bits1], dp[i][bits0]);

  int sum = 0;
  for (int bits = 0; bits < nbits; bits++)
    addmod(sum, dp[m - 1][bits]);

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