結果

問題 No.1368 サイクルの中に眠る門松列
ユーザー mmn15277198mmn15277198
提出日時 2021-01-31 14:16:26
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 72 ms / 2,000 ms
コード長 1,320 bytes
コンパイル時間 765 ms
コンパイル使用メモリ 76,432 KB
実行使用メモリ 13,984 KB
最終ジャッジ日時 2024-06-28 23:01:03
合計ジャッジ時間 2,154 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;
using ll = long long;
ll kado(int a , int b , int c){
    if(a == b || b == c || c == a){
        return 0;
    }
    if(b < a && b < c){
        return a;
    }
    if(b > a && b > c){
        return a;
    }
    return 0;
}

void solve(){
    int n;
    cin >> n;
    vector<ll> a1(n);
    for(int i = 0; i < n; i++){
        cin >> a1[i];
    }

    vector<ll> a2 = a1;
    a2.push_back(a1[0]);
    a2.erase(a2.begin());

    vector<ll> a3 = a2;
    a3.push_back(a2[0]);
    a3.erase(a3.begin());

    vector<ll> dp1(n , 0);
    dp1[2] = kado(a1[0] , a1[1] , a1[2]);
    for(int i = 3; i < n; i++){
        dp1[i] = max(dp1[i - 1] , kado(a1[i - 2] , a1[i - 1] , a1[i]) + dp1[i - 3]);
    }

    vector<ll> dp2(n , 0);
    dp2[2] = kado(a2[0] , a2[1] , a2[2]);
    for(int i = 3; i < n; i++){
        dp2[i] = max(dp2[i - 1] , kado(a2[i - 2] , a2[i - 1] , a2[i]) + dp2[i - 3]);
    }

    vector<ll> dp3(n , 0);
    dp3[2] = kado(a3[0] , a3[1] , a3[2]);
    for(int i = 3; i < n; i++){
        dp3[i] = max(dp3[i - 1] , kado(a3[i - 2] , a3[i - 1] , a3[i]) + dp3[i - 3]);
    }
    cout << max({dp1[n - 1] , dp2[n - 1] , dp3[n - 1]}) << endl;
}
int main() {
    int t;
    cin >> t;
    for(int i = 0; i < t; i++)solve();
    return 0;
}
0