oj.uz

Submission #652688

# Submission time Handle Problem Language Result Execution time Memory
652688 2022-10-23T20:48:26 Z Lobo Catfish Farm (IOI22_fish) C++17
23 / 100
 1000 ms 131112 KB 
#include "fish.h"
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define pb push_back
#define mp make_pair
#define fr first
#define sc second
#define all(x) x.begin(),x.end()

const int maxn = 1e5+10;
const int maxh = 12;
const int maxqh = 12;
const int inf = 1e18+10;

int n, m, dp[maxn][maxqh][maxqh];
vector<pair<int,int>> fish[maxn]; // fishs from column i
vector<int> hr[maxn]; // important heights from column i, the ones that can be chosen

long long max_weights(int32_t N, int32_t M, vector<int32_t> X, vector<int32_t> Y, vector<int32_t> W) {
    n = N;
    m = M;

    for(int i = 0; i < m; i++) {
        fish[X[i]+1].pb(mp(Y[i]+1,W[i]));
        hr[X[i]+1].pb(Y[i]);
    }

    for(int i = 0; i <= n; i++) {
        hr[i].pb(0);
        hr[i].pb(maxh-1);
        sort(all(hr[i]));
        hr[i].erase(unique(all(hr[i])),hr[i].end());
    }

    // dp[i][h1][h] = until i and with h[i-1] = hr[i-1][h1] and h[i] = hr[i][h-1]

    // dp[1][0][x] = 0;
    for(int i = 0; i < hr[i].size(); i++) dp[1][0][i] = 0;
    for(int i = 2; i <= n; i++) {
        for(int h1 = 0; h1 < hr[i-1].size(); h1++) {
            for(int h = 0; h < hr[i].size(); h++) {
                dp[i][h1][h] = 0;

                int plus10 = 0; // fishs that 1 catchs in 0
                for(auto aux : fish[i]) {
                    int y = aux.fr;
                    int w = aux.sc;
                    if(y > hr[i][h] && y <= hr[i-1][h1]) plus10+= w;
                }

                for(int h2 = 0; h2 < hr[i-2].size(); h2++) {
                    int plus01 = 0; // fishs that 0 catchs in 1 and had not been caught by 2

                    for(auto aux : fish[i-1]) {
                        int y = aux.fr;
                        int w = aux.sc;
                        if(y > hr[i-1][h1] && y <= hr[i][h] && y > hr[i-2][h2]) plus01+= w;
                    }
                    dp[i][h1][h] = max(dp[i][h1][h],dp[i-1][h2][h1]+plus01 + plus10);
                }
            }
        }
    }

    int ans = 0;
    for(int h = 0; h < hr[n].size(); h++) {
        for(int h1 = 0; h1 < hr[n-1].size(); h1++) {
            ans = max(ans,dp[n][h1][h]);
        }
    }

    // // dp[i][h1][h] = until i and with h[i-1] = h1 and h[i] = h

    // // dp[1][0][h] = 0;
    // for(int i = 0; i <= maxh-1; i++) dp[1][0][i] = 0;
    // for(int i = 2; i <= n; i++) {
    //     for(int h1 = 0; h1 <= maxh-1; h1++) {
    //         for(int h = 0; h <= maxh-1; h++) {
    //             dp[i][h1][h] = 0;

    //             int plus10 = 0; // fishs that 1 catchs in 0
    //             for(auto aux : fish[i]) {
    //                 int y = aux.fr;
    //                 int w = aux.sc;
    //                 if(y > h && y <= h1) plus10+= w;
    //             }

    //             for(int h2 = 0; h2 <= maxh-1; h2++) {
    //                 int plus01 = 0; // fishs that 0 catchs in 1 and had not been caught by 2

    //                 for(auto aux : fish[i-1]) {
    //                     int y = aux.fr;
    //                     int w = aux.sc;
    //                     if(y > h1 && y <= h && y > h2) plus01+= w;
    //                 }
    //                 dp[i][h1][h] = max(dp[i][h1][h],dp[i-1][h2][h1]+plus01 + plus10);
    //             }
    //         }
    //     }
    // }

    // int ans = 0;
    // for(int h = 0; h <= maxh-1; h++) {
    //     for(int h1 = 0; h1 <= maxh-1; h1++) {
    //         ans = max(ans,dp[n][h1][h]);
    //     }
    // }
    return ans;
}

Compilation message

fish.cpp: In function 'long long int max_weights(int32_t, int32_t, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:39:22: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   39 |     for(int i = 0; i < hr[i].size(); i++) dp[1][0][i] = 0;
      |                    ~~^~~~~~~~~~~~~~
fish.cpp:41:28: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   41 |         for(int h1 = 0; h1 < hr[i-1].size(); h1++) {
      |                         ~~~^~~~~~~~~~~~~~~~
fish.cpp:42:30: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   42 |             for(int h = 0; h < hr[i].size(); h++) {
      |                            ~~^~~~~~~~~~~~~~
fish.cpp:52:36: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   52 |                 for(int h2 = 0; h2 < hr[i-2].size(); h2++) {
      |                                 ~~~^~~~~~~~~~~~~~~~
fish.cpp:67:22: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   67 |     for(int h = 0; h < hr[n].size(); h++) {
      |                    ~~^~~~~~~~~~~~~~
fish.cpp:68:28: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   68 |         for(int h1 = 0; h1 < hr[n-1].size(); h1++) {
      |                         ~~~^~~~~~~~~~~~~~~~

Subtask #1
0 / 3.0

# Verdict Execution time Memory Grader output
1 Execution timed out 1072 ms 12028 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #2
0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 4948 KB Output is correct
2 Execution timed out 1092 ms 15596 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #3
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 52 ms 120780 KB Output is correct
2 Correct 55 ms 120784 KB Output is correct
3 Correct 82 ms 113632 KB Output is correct
4 Correct 77 ms 123352 KB Output is correct
5 Correct 117 ms 130980 KB Output is correct
6 Correct 113 ms 131016 KB Output is correct
7 Correct 131 ms 131112 KB Output is correct
8 Correct 128 ms 130956 KB Output is correct

Subtask #4
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 4948 KB Output is correct
2 Correct 2 ms 4948 KB Output is correct
3 Correct 3 ms 4948 KB Output is correct
4 Correct 3 ms 4948 KB Output is correct
5 Correct 3 ms 4948 KB Output is correct
6 Correct 3 ms 4948 KB Output is correct
7 Correct 3 ms 4948 KB Output is correct
8 Correct 3 ms 4948 KB Output is correct
9 Correct 4 ms 5204 KB Output is correct
10 Correct 9 ms 5460 KB Output is correct
11 Correct 5 ms 5204 KB Output is correct
12 Correct 5 ms 5432 KB Output is correct
13 Correct 3 ms 5076 KB Output is correct
14 Correct 6 ms 5332 KB Output is correct

Subtask #5
0 / 21.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 4948 KB Output is correct
2 Correct 2 ms 4948 KB Output is correct
3 Correct 3 ms 4948 KB Output is correct
4 Correct 3 ms 4948 KB Output is correct
5 Correct 3 ms 4948 KB Output is correct
6 Correct 3 ms 4948 KB Output is correct
7 Correct 3 ms 4948 KB Output is correct
8 Correct 3 ms 4948 KB Output is correct
9 Correct 4 ms 5204 KB Output is correct
10 Correct 9 ms 5460 KB Output is correct
11 Correct 5 ms 5204 KB Output is correct
12 Correct 5 ms 5432 KB Output is correct
13 Correct 3 ms 5076 KB Output is correct
14 Correct 6 ms 5332 KB Output is correct
15 Incorrect 3 ms 5332 KB 1st lines differ - on the 1st token, expected: '299', found: '298'
16 Halted 0 ms 0 KB -

Subtask #6
0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 4948 KB Output is correct
2 Correct 2 ms 4948 KB Output is correct
3 Correct 3 ms 4948 KB Output is correct
4 Correct 3 ms 4948 KB Output is correct
5 Correct 3 ms 4948 KB Output is correct
6 Correct 3 ms 4948 KB Output is correct
7 Correct 3 ms 4948 KB Output is correct
8 Correct 3 ms 4948 KB Output is correct
9 Correct 4 ms 5204 KB Output is correct
10 Correct 9 ms 5460 KB Output is correct
11 Correct 5 ms 5204 KB Output is correct
12 Correct 5 ms 5432 KB Output is correct
13 Correct 3 ms 5076 KB Output is correct
14 Correct 6 ms 5332 KB Output is correct
15 Incorrect 3 ms 5332 KB 1st lines differ - on the 1st token, expected: '299', found: '298'
16 Halted 0 ms 0 KB -

Subtask #7
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 52 ms 120780 KB Output is correct
2 Correct 55 ms 120784 KB Output is correct
3 Correct 82 ms 113632 KB Output is correct
4 Correct 77 ms 123352 KB Output is correct
5 Correct 117 ms 130980 KB Output is correct
6 Correct 113 ms 131016 KB Output is correct
7 Correct 131 ms 131112 KB Output is correct
8 Correct 128 ms 130956 KB Output is correct
9 Incorrect 123 ms 131008 KB 1st lines differ - on the 1st token, expected: '99999', found: '99998'
10 Halted 0 ms 0 KB -

Subtask #8
0 / 16.0

# Verdict Execution time Memory Grader output
1 Execution timed out 1072 ms 12028 KB Time limit exceeded
2 Halted 0 ms 0 KB -