제출 #340917

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
340917		mihai145	Visiting Singapore (NOI20_visitingsingapore)	C++14	23 / 100	237 ms	620 KiB

VisitingSingapore

#include <iostream>

using namespace std;
/*
ifstream cin("txt.in");
ofstream cout("txt.out");
*/
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;

    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;
    dp[1][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
            if(dp[0][j][0][0] != INF)
                dp[0][j + 1][0][1] = max(dp[0][j + 1][0][1], dp[0][j][0][0] + A + B);

            if(dp[0][j][0][1] != INF)
                dp[0][j + 1][0][1] = max(dp[0][j + 1][0][1], dp[0][j][0][1] + B);

            if(dp[0][j][1][0] != INF)
                dp[0][j + 1][1][1] = max(dp[0][j + 1][1][1], dp[0][j][1][0] + A + B);

            if(dp[0][j][1][1] != INF)
                dp[0][j + 1][1][1] = max(dp[0][j + 1][1][1], dp[0][j][1][1] + B);

            ///Skip day
            if(dp[0][j][0][0] != INF)
                dp[1][j][1][0] = max(dp[1][j][1][0], dp[0][j][0][0] + A + B);

            if(dp[0][j][1][0] != INF)
                dp[1][j][1][0] = max(dp[1][j][1][0], dp[0][j][1][0] + B);

            if(dp[0][j][0][1] != INF)
                dp[1][j][1][1] = max(dp[1][j][1][1], dp[0][j][0][1] + A + B);

            if(dp[0][j][1][1] != INF)
                dp[1][j][1][1] = max(dp[1][j][1][1], dp[0][j][1][1] + B);

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

                if(dp[0][j][1][0] != INF)
                    dp[1][j + 1][0][0] = max(dp[1][j + 1][0][0], dp[0][j][1][0] + v[t[j + 1]]);

                if(dp[0][j][0][1] != INF)
                    dp[1][j + 1][0][0] = max(dp[1][j + 1][0][0], dp[0][j][0][1] + v[t[j + 1]]);

                if(dp[0][j][1][1] != INF)
                    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;
                }

        dp[1][0][0][0] = 0;
    }

    cout << res << '\n';

    return 0;
}

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