結果
問題 | No.2423 Merge Stones |
ユーザー |
![]() |
提出日時 | 2023-08-12 13:59:06 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 383 ms / 4,000 ms |
コード長 | 1,285 bytes |
コンパイル時間 | 1,674 ms |
コンパイル使用メモリ | 175,780 KB |
実行使用メモリ | 6,820 KB |
最終ジャッジ日時 | 2024-11-19 16:57:14 |
合計ジャッジ時間 | 20,294 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 72 |
ソースコード
#include <bits/stdc++.h> using namespace std; int main(){ int N, K; cin >> N >> K; vector<int> A(N); for (int i = 0; i < N; i++){ cin >> A[i]; } vector<int> C(N); for (int i = 0; i < N; i++){ cin >> C[i]; } vector<long long> S(N + 1); S[0] = 0; for (int i = 0; i < N; i++){ S[i + 1] = S[i] + A[i]; } auto sum = [&](int L, int R){ if (L < R){ return S[R] - S[L]; } else { return (S[N] - S[L]) + S[R]; } }; auto match = [&](long long a, long long b){ long long ans = a & b; for (int i = 1; i <= K; i++){ ans |= a & (b << i); ans |= a & (b >> i); ans |= (a << i) & b; ans |= (a >> i) & b; } return ans; }; vector<vector<long long>> dp(N, vector<long long>(N, 0)); for (int i = 0; i < N - 1; i++){ dp[i][i + 1] = (long long) 1 << C[i]; } dp[N - 1][0] = (long long) 1 << C[N - 1]; for (int i = 2; i <= N; i++){ for (int j = 0; j < N; j++){ for (int k = j + 1; k < j + i; k++){ dp[j][(j + i) % N] |= match(dp[j][k % N], dp[k % N][(j + i) % N]); } } } long long ans = 0; for (int i = 0; i < N; i++){ for (int j = 0; j < N; j++){ if (dp[i][j] != 0){ ans = max(ans, sum(i, j)); } } } cout << ans << endl; }