Submission #785401

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

fish

#include<bits/stdc++.h>
using namespace std;

long long max_weights(int N, int M, vector<int> X, vector<int> Y, vector<int> W) {
	vector<vector<int>> go(N);
	vector<vector<pair<int, int>>> Ws(N);
	for(int i = 0; i < M; ++i) {
		++Y[i];
		if(X[i] - 1 >= 0) go[X[i]].push_back(Y[i]);
		if(X[i] + 1 < N) go[X[i] + 1].push_back(Y[i]);
		Ws[X[i]].emplace_back(Y[i], W[i]);
	}
	for(auto& u : go) {
		u.push_back(0);
		sort(begin(u), end(u));
		u.erase(unique(begin(u), end(u)), end(u));
	}
	for(auto& u : Ws) {
		sort(begin(u), end(u));
		u.erase(unique(begin(u), end(u)), end(u));
		for(int i = 1; i < (int)u.size(); ++i) {
			u[i].second += u[i - 1].second;
		}
	}
	auto sum = [&](int x, int y) -> long long {
		if(x < 0 || x >= N) return 0;
		if(y == 0) return 0;
		auto it = lower_bound(begin(Ws[x]), end(Ws[x]), make_pair(y, 0));
		if((*it).first != y) return 0;
		return (*it).second;
	};
	vector<map<int, map<int, long long>>> dp(N);
	long long ans = 0;
	for(int i = 0; i < N; ++i) {
		for(int ci = 0; ci < (int)go[i].size(); ++ci) {
			int u = go[i][ci];
			if(i - 1 >= 0) for(int pi = 0; pi < (int)go[i - 1].size(); ++pi) {
				int v = go[i - 1][pi];
				if(i - 2 >= 0) for(int ppi = 0; ppi < (int)go[i - 2].size(); ++ppi) {
					int y = go[i - 2][ppi];
					long long pl = dp[i - 1][pi][ppi] + sum(i + 1, u) - sum(i, min(u, v)) + max(0LL, sum(i - 1, u) - sum(i - 1, max(v, y)));
					dp[i][ci][pi] = max(dp[i][ci][pi], pl);
					ans = max(ans, dp[i][ci][pi]);
				} else {
					dp[i][ci][pi] = dp[i - 1][pi][0] + sum(i + 1, u) - sum(i, min(u, v)) + max(0LL, sum(i - 1, u) - sum(i - 1, v));
					ans = max(ans, dp[i][ci][pi]);
				}
			} else {
				if(u == 0) continue;
				dp[i][ci][0] = sum(i - 1, u) + sum(i + 1, u);
				ans = max(ans, dp[i][ci][0]);
			}
		}
	}
	return ans;
}

int main() {
	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/ccZFTvv0.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccJvzgC3.o:fish.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status