Submission #637644

#	Time	Username	Problem	Language	Result	Execution time	Memory
637644		someone	Catfish Farm (IOI22_fish)	C++17	Compilation error	0 ms	0 KiB

fish

//#include "fish.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;

struct Pier {
    ll x, y, fishL = 0, fishAct = 0, fishR = 0, val[2] = {0, 0};
};

struct Fish {
    bool req;
    ll y, pds;
};

const ll N = 1e5 + 42, INF = 1e15 + 42;

ll sz[N];
map<ll, ll> nb[N];
vector<Fish> fish[N];
vector<Pier> pier[N];

ll max_weights(int n, int M, vector<int> X, vector<int> Y, vector<int> W) {
    for(int i = 0; i < M; i++) {
        X[i] += 2;
        Y[i] += 1;
        fish[X[i]].push_back({false, Y[i], W[i]});
        for(int j = X[i]-2; j < X[i]+3; j++)
            fish[j].push_back({true, Y[i], 0});
        pier[X[i]].push_back({X[i], Y[i]});
        pier[X[i]-1].push_back({X[i], Y[i]});
        pier[X[i]+1].push_back({X[i], Y[i]});
    }

    for(int i = 0; i < N; i++) {
        sort(fish[i].begin(), fish[i].end(),
        [](Fish a, Fish b) {
            if(a.y == b.y) {
                if(a.req == b.req)
                    return false;
                return b.req;
            }
            return a.y < b.y;
        });
        ll sum = 0;
        for(Fish f : fish[i]) {
            if(f.req) {
                nb[i][f.y] = sum;
            } else {
                sum += f.pds;
            }
        }
    }
    for(int i = 0; i < N; i++) {
        for(Pier& p : pier[i]) {
            p.fishL = nb[i-1][p.y];
            p.fishAct = nb[i][p.y];
            p.fishR = nb[i+1][p.y];
        }
    }

    for(int i = 2; i < n+2; i++)
        sort(pier[i].begin(), pier[i].end(),
        [](Pier a, Pier b) {
            return a.y < b.y;
        });
    for(int i = 0; i < n+2; i++)
        sz[i] = (ll)pier[i].size();
    
    ll maxi = 0;
    for(int i = 2; i < n+2; i++) {
        for(Pier& p : pier[i]) {
            p.val[0] = maxi + p.fishL - p.fishAct;
            p.val[1] = maxi + p.fishL + p.fishR;
        }

        ll id[2] = {0, 0}, maxiLL = -INF, maxiL = -INF;
        for(Pier& p : pier[i]) {
            while(id[0] < sz[i-1] && pier[i-1][id[0]].y <= p.y) {
                maxiL = max(maxiL, pier[i-1][id[0]].val[0]);
                id[0]++;
            }
            while(id[1] < sz[i-2] && pier[i-2][id[1]].y <= p.y) {
                maxiLL = max(maxiLL, pier[i-2][id[1]].val[1] - pier[i-2][id[1]].fishR);
                id[1]++;
            }
            p.val[0] = max(p.val[0], max(maxiL - p.fishAct, maxiLL + p.fishR) + p.fishL);
            p.val[1] = max(p.val[1], max(maxiL, maxiLL) + p.fishL + p.fishR);
        }
        id[0] = sz[i-1]-1;
        id[1] = sz[i-2]-1;
        maxiLL = -INF, maxiL = -INF;
        reverse(pier[i].begin(), pier[i].end());

        for(Pier& p : pier[i]) {
            while(id[0] >= 0 && pier[i-1][id[0]].y >= p.y) {
                maxiL = max(maxiL, pier[i-1][id[0]].val[1]);
                id[0]--;
            }
            while(id[1] >= 0 && pier[i-2][id[1]].y >= p.y) {
                maxiLL = max(maxiLL, pier[i-2][id[1]].val[1]);
                id[1]--;
            }
            p.val[0] = max(p.val[0], maxiLL - p.fishAct);
            p.val[1] = max(p.val[1], max(maxiLL, maxiL - p.fishAct) + p.fishR);
        }

        reverse(pier[i].begin(), pier[i].end());

        for(Pier p : pier[i-1])
            maxi = max(maxi, max(p.val[0], p.val[1]));
    }
    return maxi;
}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    
    int n, m;
    cin >> n >> m;
    vector<int> x(m), y(m), w(m);
    for(int i = 0; i < m; i++)
        cin >> x[i] >> y[i] >> w[i];
    cout << max_weights(n, m, x, y, w);
}

Compilation message (stderr)

/usr/bin/ld: /tmp/ccyeQUEj.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccgdUZDf.o:fish.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status