Submission #340903

#	Time	Username	Problem	Language	Result	Execution time	Memory
340903		mihai145	Visiting Singapore (NOI20_visitingsingapore)	C++14	0 / 100	204 ms	620 KiB

VisitingSingapore

#include <iostream>

using namespace std;

const int INF = 1e9;

const int KMAX = 1000;
const int NMAX = 5000;

int K, N, M, A, B;
int dp[2][NMAX + 5][2][2];

int v[KMAX + 5], s[NMAX + 5], t[NMAX + 5];

int main()
{

    cin >> K >> N >> M >> A >> B;
    A = -A;
    B = -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];

    for(int i = 0; i < 2; i++)
        for(int j = 0; j <= M; j++)
            for(int x = 0; x < 2; x++)
                for(int y = 0; y < 2; y++)
                    dp[i][j][x][y] = -INF;

    dp[0][0][0][0] = 0;

    int res = A + M * B;

    for(int i = 1; i <= N; i++)
    {
        for(int j = 0; j <= M - 1; j++)
        {
            ///Skip concert
            dp[0][j + 1][0][1] = max(dp[0][j + 1][0][1], dp[0][j][0][0] - A - B);
            dp[0][j + 1][0][1] = max(dp[0][j + 1][0][1], dp[0][j][0][1] - B);
            dp[0][j + 1][1][1] = max(dp[0][j + 1][1][1], dp[0][j][1][0] - A - B);
            dp[0][j + 1][1][1] = max(dp[0][j + 1][1][1], dp[0][j][1][1] - B);

            ///Skip day
            dp[1][j][1][0] = max(dp[1][j][1][0], dp[0][j][0][0] - A - B);
            dp[1][j][1][0] = max(dp[1][j][1][0], dp[0][j][1][0] - B);
            dp[1][j][1][1] = max(dp[1][j][1][1], dp[0][j][0][1] - A - B);
            dp[1][j][1][1] = max(dp[1][j][1][1], dp[0][j][1][1] - B);

            ///Attend
            if(s[i] == t[j + 1])
            {
                dp[1][j + 1][0][0] = max(dp[1][j + 1][0][0], dp[0][j][0][0] + v[t[j + 1]]);
                dp[1][j + 1][0][0] = max(dp[1][j + 1][0][0], dp[0][j][1][0] + v[t[j + 1]]);
                dp[1][j + 1][0][0] = max(dp[1][j + 1][0][0], dp[0][j][0][1] + v[t[j + 1]]);
                dp[1][j + 1][0][0] = max(dp[1][j + 1][0][0], dp[0][j][1][1] + v[t[j + 1]]);
            }
        }

        res = max(res, dp[0][M][0][0]);
        res = max(res, dp[0][M][0][1]);
        res = max(res, dp[0][M][1][0]);
        res = max(res, dp[0][M][1][1]);
        res = max(res, dp[1][M][0][0]);
        res = max(res, dp[1][M][0][1]);
        res = max(res, dp[1][M][1][0]);
        res = max(res, dp[1][M][1][1]);

        for(int j = 0; j <= M; j++)
            for(int x = 0; x < 2; x++)
                for(int y = 0; y < 2; y++)
                {
                    dp[0][j][x][y] = dp[1][j][x][y];
                    dp[1][j][x][y] = -INF;
                }
    }

    cout << res << '\n';

    return 0;
}

• Subtask 1: 0 / 4
• Subtask 2: 0 / 6
• Subtask 3: 0 / 12
• Subtask 4: 0 / 7
• Subtask 5: 0 / 8
• Subtask 6: 0 / 13
• Subtask 7: 0 / 50