This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define fixbug false
const ll INF = (ll)1e18+7;
const int maxn = 5e3;
int K , n , m ;
ll A , B;
int s[maxn+2] , t[maxn+2] ;
ll v[maxn+2];
namespace subtaskfull{
bool check(){
return true;
}
const ll INF = (ll)1e18+7;
ll dp[maxn+2][2][2]; // FOR the layer , SKIP S , SKIP T
ll tam[maxn+2][2][2];
void main_code(){
memset(dp,-0x3f,sizeof dp);
ll ans = -INF;
for (int i = 0; i <= n; ++i) dp[i][1][1] = 0; // START state
for (int layer = 1; layer <= m; ++layer){
for (int i = 0; i <= n; ++i){
for (int match1 = 0; match1 <= 1; ++match1){
for (int match2 = 0; match2 <= 1; ++match2){
tam[i][match1][match2] = dp[i][match1][match2];
dp[i][match1][match2] = -INF;
if (i==0) dp[i][match1][0] = max(dp[i][match1][0],tam[i][match1][match2] + B + A * match2);
}
}
}
for (int i = 1; i <= n; ++i){
for (int match1 = 0; match1 <= 1; ++match1)
for (int match2 = 0; match2 <= 1; ++match2)
{
if (s[i]==t[layer]){
dp[i][1][1] = max({
dp[i][1][1] ,
tam[i - 1][match1][match2] + v[t[layer]]
});
}
// SKIP S
dp[i][0][match2] = max({
dp[i][0][match2] ,
dp[i-1][match1][match2] + B + A * match1
});
// SKIP T
dp[i][match1][0] = max({
dp[i][match1][0] ,
tam[i][match1][match2] + B + A * match2
});
}
}
}
for (int i = 0; i <= n; ++i){
for (int match1 = 0; match1 <= 1; ++match1){
for (int match2 = 0; match2 <= 1; ++match2){
ans = max(ans , dp[i][match1][match2]);
}
}
}
cout << ans << '\n';
}
}
int main(){
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cin >> K >> n >> m >> A >> B;
for (int i = 1; i <= K; ++i) cin >> v[i];
for (int i = 1; i <= n; ++i) cin >> s[i];
for (int i = 1; i <= m; ++i) cin >> t[i];
if (subtaskfull::check()){
subtaskfull::main_code();
exit(0);
}
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |