結果

問題 No.1897 Sum of 2nd Max
ユーザー polylogK
提出日時 2021-12-12 21:30:24
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 611 bytes
コンパイル時間 2,102 ms
コンパイル使用メモリ 198,200 KB
最終ジャッジ日時 2025-01-26 09:20:03
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 TLE * 1
other AC * 1 TLE * 29
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:6:24: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
    6 |         int n, k; scanf("%d%d", &n, &k);
      |                   ~~~~~^~~~~~~~~~~~~~~~

ソースコード

diff # 

#include <bits/stdc++.h>

constexpr int MOD = 998244353;

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

	int ans = 0;
	std::vector<int> a;
	auto dfs = [&](auto&& dfs, int i) -> void {
		if(i == n) {
			int max1 = std::max(a[0], a[1]);
			int max2 = std::min(a[0], a[1]);
			for(int i = 2; i < n; i++) {
				if(a[i] > max1) {
					max2 = max1;
					max1 = a[i];
				} else if(a[i] > max2) {
					max2 = a[i];
				}
			}
			ans = (ans + max2) % MOD;
			return;
		}

		for(int d = 1; d <= k; d++) {
			a.push_back(d);
			dfs(dfs, i + 1);
			a.pop_back();
		}
	}; dfs(dfs, 0);

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