結果
問題 | 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;}