oj.uz

답안 #1042044

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1042044 2024-08-02T13:11:14 Z VMaksimoski008 메기 농장 (IOI22_fish) C++17
12 / 100
 49 ms 10836 KB 
#include "fish.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
// #define int64_t long long

ll max_weights(int N, int M, vector<int> X, vector<int> Y, vector<int> W) {
    int mask = 0, mx = 0;
    for(int64_t i=0; i<M; i++) mask |= (1 << (X[i] % 2));
    for(int64_t i=0; i<M; i++) mx = max(mx, Y[i]);

    if(mask == 1) {
        ll ans = 0;
        for(auto &x : W) ans += x;
        return ans;
    }

    if(mx == 0) {
        vector<ll> val(N+5), dp(N+5);
        for(int64_t i=0; i<M; i++) val[X[i]] = W[i];

        ll ans = 0;
        vector<ll> MX(N+5); MX[0] = val[1];
        for(int64_t i=1; i<N; i++) {
            dp[i] = max(dp[i-1], val[i-1]);
            if(i >= 3) dp[i] = max(dp[i], MX[i-3] + val[i-1]);
            if(i >= 2) dp[i] = max(dp[i], dp[i-2] + val[i-1]);
            MX[i] = max(MX[i-1], dp[i] + val[i+1]);
        }

        for(int64_t i=0; i<N; i++) ans = max(ans, dp[i] + val[i+1]);
        return ans;
    }

    if(mx <= 8) {
        ll val[N+5][mx+5], dp[N+5][mx+5][mx+5];
        memset(val, 0, sizeof(val)); memset(dp, 0, sizeof(dp));
        for(int64_t i=0; i<M; i++) val[X[i]][Y[i]] = W[i];
        for(int64_t i=0; i<N; i++)
            for(int64_t j=1; j<=mx; j++) val[i][j] += val[i][j-1];

        for(int64_t j=0; j<=mx+1; j++) {
            for(int64_t k=0; k<=mx+1; k++) {
                if(j > k) dp[1][j][k] = val[0][j-1] - (k ? val[0][k-1] : 0);
                else if(j < k) dp[1][j][k] = val[1][k-1] - (j ? val[1][j-1] : 0);
            }
        }

        for(int64_t i=2; i<N; i++) {
            for(int64_t j=0; j<=mx+1; j++) {
                for(int64_t k=0; k<=mx+1; k++) {
                    if(j > k) dp[i][j][k] = (j ? val[i-1][j-1] : 0) - (k ? val[i-1][k-1] : 0);
                    if(j < k) dp[i][j][k] = (k ? val[i][k-1] : 0) - (j ? val[i][j-1] : 0);
                    if(k) dp[i][j][k] += val[i-2][k-1];

                    for(int64_t l=0; l<=mx+1; l++) {
                        if(l <= k && k <= j) {
                            dp[i][j][k] = max(dp[i][j][k], dp[i-1][k][l] + (j ? val[i-1][j-1] : 0) - (k ? val[i-1][k-1] : 0));
                        } 

                        if(l >= k && k >= j) {
                            dp[i][j][k] = max(dp[i][j][k], dp[i-1][k][l] + (k ? val[i][k-1] : 0) - (j ? val[i][j-1] : 0));
                        }

                        if(l <= k && k >= j) {
                            dp[i][j][k] = max(dp[i][j][k], dp[i-1][k][l] + (k ? val[i][k-1] : 0) - (j ? val[i][j-1] : 0));
                        }

                        if(l >= k && k <= j) {
                            if(l >= j) {
                                dp[i][j][k] = max(dp[i][j][k], dp[i-1][k][l]);
                            } else {
                                dp[i][j][k] = max(dp[i][j][k], dp[i-1][k][l] + (j ? val[i-1][j-1] : 0) - (l ? val[i-1][l-1] : 0));
                            }
                        }
                    }
                }
            }
        }

        ll ans = 0;
        // // cout << dp[3][3][2] << '\n';

        // for(int64_t i=0; i<N; i++) {
        //     for(int64_t j=0; j<=mx+1; j++) {
        //         for(int64_t k=0; k<=mx+1; k++) {
        //             cout << i << " " << j << " " << k << ": " << dp[i][j][k] << '\n';
        //         }
        //     }
        // }

        for(int64_t i=0; i<N; i++)
            for(int64_t j=0; j<=mx+1; j++)
                for(int64_t k=0; k<=mx+1; k++) ans = max(ans, dp[i][j][k] + val[i+1][j]);
        return ans;
    }


    return 0;
}

Subtask #1
3.0 / 3.0

# 결과 실행 시간 메모리 Grader output
1 Correct 12 ms 3420 KB Output is correct
2 Correct 14 ms 3600 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 49 ms 10684 KB Output is correct
6 Correct 49 ms 10836 KB Output is correct

Subtask #2
0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Incorrect 27 ms 5656 KB 1st lines differ - on the 1st token, expected: '40604614618209', found: '0'
3 Halted 0 ms 0 KB -

Subtask #3
9.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 2652 KB Output is correct
3 Correct 9 ms 4500 KB Output is correct
4 Correct 7 ms 3932 KB Output is correct
5 Correct 16 ms 5972 KB Output is correct
6 Correct 13 ms 5724 KB Output is correct
7 Correct 15 ms 6116 KB Output is correct
8 Correct 16 ms 5976 KB Output is correct

Subtask #4
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 344 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Incorrect 1 ms 348 KB 1st lines differ - on the 1st token, expected: '216624184325', found: '217189184593'
10 Halted 0 ms 0 KB -

Subtask #5
0 / 21.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 344 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Incorrect 1 ms 348 KB 1st lines differ - on the 1st token, expected: '216624184325', found: '217189184593'
10 Halted 0 ms 0 KB -

Subtask #6
0 / 17.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 344 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Incorrect 1 ms 348 KB 1st lines differ - on the 1st token, expected: '216624184325', found: '217189184593'
10 Halted 0 ms 0 KB -

Subtask #7
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 2652 KB Output is correct
3 Correct 9 ms 4500 KB Output is correct
4 Correct 7 ms 3932 KB Output is correct
5 Correct 16 ms 5972 KB Output is correct
6 Correct 13 ms 5724 KB Output is correct
7 Correct 15 ms 6116 KB Output is correct
8 Correct 16 ms 5976 KB Output is correct
9 Incorrect 14 ms 3672 KB 1st lines differ - on the 1st token, expected: '99999', found: '0'
10 Halted 0 ms 0 KB -

Subtask #8
0 / 16.0

# 결과 실행 시간 메모리 Grader output
1 Correct 12 ms 3420 KB Output is correct
2 Correct 14 ms 3600 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 49 ms 10684 KB Output is correct
6 Correct 49 ms 10836 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Incorrect 27 ms 5656 KB 1st lines differ - on the 1st token, expected: '40604614618209', found: '0'
9 Halted 0 ms 0 KB -