結果

問題 No.572 妖精の演奏
コンテスト
ユーザー rpy3cpp
提出日時 2017-10-07 03:27:06
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 13 ms / 2,000 ms
コード長 1,643 bytes
コンパイル時間 1,896 ms
コンパイル使用メモリ 179,600 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-17 03:29:00
合計ジャッジ時間 2,747 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

long long solve(long long n, int m, const vector<vector<int>> & A){
    vector<vector<vector<long long>>> dp(m, vector<vector<long long>>(m + 1, vector<long long>(m, 0)));
    for (int s = 0; s < m; ++s){
        for (int d = 1; d < m + 1; ++d){
            for (int g = 0; g < m; ++g){
                if (d == 1){
                    dp[s][d][g] = A[s][g];
                    continue;
                };
                for (int v = 0; v < m; ++v){
                    dp[s][d][g] = max(dp[s][d][g], dp[s][d - 1][v] + A[v][g]);
                }
            }
        }
    }
    vector<vector<long long>> tail(m, vector<long long>(m + 1, 0));
    for (int s = 0; s < m; ++s){
        for (int d = 1; d < m + 1; ++d){
            tail[s][d] = *max_element(begin(dp[s][d]), end(dp[s][d]));
        }
    }
    --n;
    long long record = 0;
    for (int s = 0; s < m; ++s){
        for (int d1 = 0; d1 < m and n - d1 >= 0; ++d1){
            for (int v = 0; v < m; ++v){
                for (int d2 = 1; d2 < m + 1; ++d2){
                    long long cycle = (n - d1)/(long long)d2;
                    auto d3 = n - d1 - cycle * d2;
                    long long score = dp[s][d1][v] + dp[v][d2][v] * cycle + tail[v][d3];
                    if (score > record) record = score;
                }
            }
        }
    }
    return record;
}



int main() {
    long long n;
    cin >> n;
    int m;
    cin >> m;
    vector<vector<int>> A(m, vector<int>(m, 0));
    for (auto & row : A){
        for (auto & v : row) cin >> v;
    }
    cout << solve(n, m, A) << endl;
    return 0;
}
0