Submission #785380

#TimeUsernameProblemLanguageResultExecution timeMemory
785380khshgCatfish Farm (IOI22_fish)C++17
Compilation error
0 ms0 KiB
#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; 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; }

Compilation message (stderr)

fish.cpp: In function 'long long int max_weights(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:37:9: warning: unused variable 'v' [-Wunused-variable]
   37 |     int v = go[i - 1][pi];
      |         ^
fish.cpp:55:1: error: expected '}' at end of input
   55 | }
      | ^
fish.cpp:4:82: note: to match this '{'
    4 | long long max_weights(int N, int M, vector<int> X, vector<int> Y, vector<int> W) {
      |                                                                                  ^
fish.cpp:5:26: warning: control reaches end of non-void function [-Wreturn-type]
    5 |  vector<vector<int>> go(N);
      |                          ^