제출 #1286415

#제출 시각아이디문제언어결과실행 시간메모리
1286415jackofall718Visiting Singapore (NOI20_visitingsingapore)C++20
20 / 100
225 ms196348 KiB
#include <bits/stdc++.h>
#include <chrono>
#define ll long long int
#define endl '\n'
#define vn vector<ll>
#define vi vector<pair <ll,ll>>

using namespace std;
using namespace std::chrono;
const int MAX_N = 1e9 + 7;
#define pii pair<ll,ll>
const ll INF = 0x3f3f3f3f3f3f3f3f;

#define pb push_back  
#define srt(vp) sort(vp.begin(), vp.end()) 

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    auto start = high_resolution_clock::now();

    ll k,n,m,a,b;
    cin>>k>>n>>m>>a>>b;
    vn v(k+1);//v
    for (int i=1;i<=k;i++)cin>>v[i];
    vn s(n+1);//s
    for (int i=1;i<=n;i++)cin>>s[i];
    vn t(m+1);//t
    for (int i=1;i<=m;i++)cin>>t[i];
    map <ll,vn> pos;
    set <ll> s2;
    for (int i=1;i<=m;i++){
        pos [t[i]].pb(i);
        s2.insert(t[i]);
    }

    vector <vn> dp(n+1,vn(m+1,-MAX_N));
    for (int i=0;i<=n;i++)dp[i][0]=0;
    ll ans = a+m*b;

    vn optimal (m+1,-MAX_N);

    for (int i=1;i<=n;i++){
        ll temp = -MAX_N;

        if (s2.count(s[i])){
            for (auto &x:pos[s[i]]){
                dp[i][x]=max(dp[i][x],dp[i-1][x-1]+v[s[i]]);
                dp[i][x]=max(dp[i][x],a+v[s[i]]);
            }
        }

        for (int j=1;j<=m;j++){
            // ↓↓↓ added b-penalty handling ↓↓↓
            optimal[j]=max(optimal[j]-b,dp[i][j]);  // skipping S[i] ⇒ −b
            dp[i][j]=max(dp[i][j],optimal[j]+a);    // block skip ⇒ −a
            temp=max(temp-b,dp[i][j]);              // skipping T[j] ⇒ −b
            dp[i][j]=max(dp[i][j],temp+a);          // block skip ⇒ −a
        }

        for (int j=1;j<=m;j++){
            ans=max(ans,dp[i][j]+(j!=m)*(a+(m-j)*b));
        }
    }

    cout<<ans<<endl;

    auto stop = high_resolution_clock::now();
    auto duration = duration_cast<microseconds>(stop - start);
    //cout << "Time taken by function: " << duration.count() << " microseconds" << endl;

    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...