oj.uz

Submission #994306

# Submission time Handle Problem Language Result Execution time Memory
994306 2024-06-07T10:37:11 Z thinknoexit Catfish Farm (IOI22_fish) C++17
3 / 100
 124 ms 41068 KB 
#include "fish.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
vector<pair<int, int>> pos[100100];
struct DP {
    ll in, de;
    DP() {
        in = de = 0;
    }
} dp[1001000];
int w[1001000];
int n, m;
ll max_weights(int N, int M, vector<int> X, vector<int> Y, vector<int> W) {
    n = N, m = M;
    for (int i = 0;i < m;i++) {
        w[i + 1] = W[i];
        pos[X[i] + 1].push_back({ Y[i] + 1, i + 1 });
    }
    int idx = m, p;
    ll mx, sum;
    for (int i = 2;i <= n;i++) {
        sort(pos[i].begin(), pos[i].end());
        if (pos[i].empty() || pos[i][0].first != 1) pos[i].push_back({ 1, ++idx });
        pos[i].push_back({ 0, ++idx });
        pos[i].push_back({ n + 1, ++idx });
        sort(pos[i].begin(), pos[i].end());
        // in -> in
        {
            p = 0;
            mx = 0;
            for (int j = 0;j < (int)pos[i].size();j++) {
                while (p < (int)pos[i - 1].size() && pos[i - 1][p].first <= pos[i][j].first - 1) {
                    int prev = pos[i - 1][p++].second;
                    mx = max(mx, dp[prev].in) + w[prev];
                }
                dp[pos[i][j].second].in = mx;
            }
        }
        // de -> de , in -> de
        {
            p = pos[i - 1].size() - 1;
            mx = 0;
            for (int j = pos[i].size() - 1;j >= 0;j--) {
                while (p >= 0 && pos[i - 1][p].first > pos[i][j].first) {
                    int prev = pos[i - 1][p--].second;
                    mx = max({ mx, dp[prev].in, dp[prev].de });
                }
                mx += w[pos[i][j].second];
                dp[pos[i][j].second].de = mx;
            }
        }
        // in -> de, de -> in
        if (i >= 3) {
            // h[i-2] <= h[i]
            {
                int mid = 0;
                p = 0;
                mx = 0, sum = 0;
                for (int j = 0;j < (int)pos[i].size();j++) {
                    while (p < (int)pos[i - 2].size() && pos[i - 2][p].first <= pos[i][j].first) {
                        int prev = pos[i - 2][p++].second;
                        mx = max({ mx ,dp[prev].in, dp[prev].de });
                    }
                    while (mid < (int)pos[i - 1].size() && pos[i - 1][mid].first <= pos[i][j].first - 1) {
                        sum += w[pos[i - 1][mid++].second];
                    }
                    dp[pos[i][j].second].in = max(dp[pos[i][j].second].in, mx + sum);
                }
            }
            // h[i-2] >= h[i]
            {
                int mid = pos[i - 1].size() - 1;
                mx = 0, sum = 0;
                for (int j = 0;j <= mid;j++) sum += w[pos[i - 1][j].second];
                p = pos[i - 2].size() - 1;
                for (int j = pos[i].size() - 1;j >= 0;j--) {
                    while (p >= 0 && pos[i - 2][p].first >= pos[i][j].first) {
                        int prev = pos[i - 2][p].second;
                        int now = pos[i - 2][p].first - 1;
                        while (mid >= 0 && pos[i - 1][mid].first > now) {
                            sum -= w[pos[i - 1][mid--].second];
                        }
                        mx = max({ mx, dp[prev].in + sum, dp[prev].de + sum });
                        p--;
                    }
                    dp[pos[i][j].second].in = max(dp[pos[i][j].second].in, mx);
                }
            }
        }
    }
    ll ans = 0;
    for (int i = 1;i <= idx;i++) {
        ans = max(ans, dp[i].in);
        ans = max(ans, dp[i].de);
    }
    return ans;
}

Subtask #1
3.0 / 3.0

# Verdict Execution time Memory Grader output
1 Correct 29 ms 29896 KB Output is correct
2 Correct 37 ms 31176 KB Output is correct
3 Correct 19 ms 24668 KB Output is correct
4 Correct 16 ms 24816 KB Output is correct
5 Correct 124 ms 39744 KB Output is correct
6 Correct 116 ms 41068 KB Output is correct

Subtask #2
0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 20060 KB Output is correct
2 Incorrect 53 ms 33088 KB 1st lines differ - on the 1st token, expected: '40604614618209', found: '40479197388443'
3 Halted 0 ms 0 KB -

Subtask #3
0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 18 ms 24664 KB Output is correct
2 Correct 17 ms 24664 KB Output is correct
3 Incorrect 30 ms 29016 KB 1st lines differ - on the 1st token, expected: '21261825233649', found: '21049836049923'
4 Halted 0 ms 0 KB -

Subtask #4
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 4 ms 20056 KB Output is correct
2 Correct 3 ms 20060 KB Output is correct
3 Correct 3 ms 20124 KB Output is correct
4 Correct 3 ms 20148 KB Output is correct
5 Correct 3 ms 20056 KB Output is correct
6 Correct 3 ms 20060 KB Output is correct
7 Correct 3 ms 20060 KB Output is correct
8 Correct 3 ms 20060 KB Output is correct
9 Incorrect 4 ms 20060 KB 1st lines differ - on the 1st token, expected: '216624184325', found: '216535146944'
10 Halted 0 ms 0 KB -

Subtask #5
0 / 21.0

# Verdict Execution time Memory Grader output
1 Correct 4 ms 20056 KB Output is correct
2 Correct 3 ms 20060 KB Output is correct
3 Correct 3 ms 20124 KB Output is correct
4 Correct 3 ms 20148 KB Output is correct
5 Correct 3 ms 20056 KB Output is correct
6 Correct 3 ms 20060 KB Output is correct
7 Correct 3 ms 20060 KB Output is correct
8 Correct 3 ms 20060 KB Output is correct
9 Incorrect 4 ms 20060 KB 1st lines differ - on the 1st token, expected: '216624184325', found: '216535146944'
10 Halted 0 ms 0 KB -

Subtask #6
0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 4 ms 20056 KB Output is correct
2 Correct 3 ms 20060 KB Output is correct
3 Correct 3 ms 20124 KB Output is correct
4 Correct 3 ms 20148 KB Output is correct
5 Correct 3 ms 20056 KB Output is correct
6 Correct 3 ms 20060 KB Output is correct
7 Correct 3 ms 20060 KB Output is correct
8 Correct 3 ms 20060 KB Output is correct
9 Incorrect 4 ms 20060 KB 1st lines differ - on the 1st token, expected: '216624184325', found: '216535146944'
10 Halted 0 ms 0 KB -

Subtask #7
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 18 ms 24664 KB Output is correct
2 Correct 17 ms 24664 KB Output is correct
3 Incorrect 30 ms 29016 KB 1st lines differ - on the 1st token, expected: '21261825233649', found: '21049836049923'
4 Halted 0 ms 0 KB -

Subtask #8
0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 29 ms 29896 KB Output is correct
2 Correct 37 ms 31176 KB Output is correct
3 Correct 19 ms 24668 KB Output is correct
4 Correct 16 ms 24816 KB Output is correct
5 Correct 124 ms 39744 KB Output is correct
6 Correct 116 ms 41068 KB Output is correct
7 Correct 3 ms 20060 KB Output is correct
8 Incorrect 53 ms 33088 KB 1st lines differ - on the 1st token, expected: '40604614618209', found: '40479197388443'
9 Halted 0 ms 0 KB -