#include <cstdio>
#ifdef DEBUG
#define D(X) X
#else
#define D(X)
#endif
#include <bits/stdc++.h>
#define F first
#define S second
#define ll long long
#define pi 3.14159265359
#define pub push_back
#define pob pop_back
#define pari pair<int,int>
#define parli pair<long, long>
using namespace std;
const int INF = 1000 * 1000 * 1000;
const int mod = 1000 * 1000 * 1000 + 7;
ll max_weights(int n, int m, vector<int> x, vector<int> y, vector<int> w)
{
vector<vector<ll>> cnt2(3, vector<ll>(n+1, 0));
for(int i=0; i<m; i++) {
if(x[i]%2==1) {
odd = 1;
}
cnt2[x[i]][y[i]+1] += w[i];
all += w[i];
mx = max(mx, x[i]);
}
if(!odd) {
return all;
}
for(int i=0; i<2; i++) {
for(int j=1; j<=n; j++) {
cnt2[i][j] += cnt2[i][j-1];
}
}
if(mx<=1) {
if(n==2) {
return max(cnt2[0][n], cnt2[1][n]);
} else
{
ll res = cnt2[1][n], best = res;
for(int j=1; j<=n; j++) {
res = max(res, res+cnt2[0][j]-cnt2[1][j]);
}
return res;
}
}
vector<vector<ll>> cnt(n + 1, vector<ll>(n+1, 0));
int odd = 0, mx = 0;
ll all = 0;
for(int i=0; i<m; i++) {
cnt[x[i]][y[i]+1] += w[i];
}
for(int i=0; i<n; i++) {
for(int j=1; j<=n; j++) {
cnt[i][j] += cnt[i][j-1];
}
}
vector<vector<ll>> pref(n+1, vector<ll>(n+1, 0));
vector<vector<ll>> suf(n+1, vector<ll>(n+1, 0));
vector<vector<vector<ll>>> dp(n+1, vector<vector<ll>>(n+1, vector<ll>(2, 0)));
ll res = 0;
for(int i=1; i<n; i++) {
if(i==1) {
pref[i-1][0] = 0;
} else
{
pref[i-1][0] = dp[i-2][0][0];
}
for(int j=1; j<=n; j++) {
pref[i-1][j] = max(pref[i-1][j-1], dp[i-1][j][1] - cnt[i-1][j]);
}
suf[i-1][n] = cnt[i][n] + max(dp[i-1][n][0], dp[i-1][n][1]);
for(int j=n-1; j>=1; j--) {
suf[i-1][j] = max(suf[i-1][j+1], max(dp[i-1][j][0], dp[i-1][j][1]) + cnt[i][j]);
}
dp[i][0][0] = dp[i][0][1] = max(suf[i-1][1], dp[i-1][0][0]);
for(int j=1; j<=n; j++) {
dp[i][j][0] = suf[i-1][j] - cnt[i][j];
dp[i][j][1] = cnt[i-1][j] + pref[i-1][j];
if(i>1) {
dp[i][j][1] = max(dp[i][j][1], cnt[i-1][j]+pref[i-2][j]);
dp[i][j][1] = max(dp[i][j][1], suf[i-2][j]);
}
//res = max(res, max(dp[i][j][0], dp[i][j][1]));
}
}
for(int i=0; i<n; i++)
for(int j=0; j<=n; j++) {
for(int bin=0; bin<2; bin++)
res = max(res, dp[i][j][bin]);
}
return res;
}
/*
int main()
{
ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
D(freopen("input.txt","r",stdin);)
//D(freopen("ouput.txt","w",stdout);)
int t = 1;
//cin >> t;
loop:
while(t--)
{
int n, m;
cin >> n >> m;
vector<int> x(m), y(m), w(m);
for(int i=0; i<m; i++) {
cin >> x[i];
}
for(int i=0; i<m; i++) {
cin >> y[i];
}
for(int i=0; i<m; i++) {
cin >> w[i];
}
ll res = max_weights(n, m, x, y, w);
cout << res;
}
return 0;
}
*/
Compilation message
fish.cpp: In function 'long long int max_weights(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:27:13: error: 'odd' was not declared in this scope
27 | odd = 1;
| ^~~
fish.cpp:30:9: error: 'all' was not declared in this scope; did you mean 'std::filesystem::perms::all'?
30 | all += w[i];
| ^~~
| std::filesystem::perms::all
In file included from /usr/include/c++/10/filesystem:44,
from /usr/include/x86_64-linux-gnu/c++/10/bits/stdc++.h:129,
from fish.cpp:7:
/usr/include/c++/10/bits/fs_fwd.h:148:7: note: 'std::filesystem::perms::all' declared here
148 | all = 0777,
| ^~~
fish.cpp:31:9: error: 'mx' was not declared in this scope; did you mean 'm'?
31 | mx = max(mx, x[i]);
| ^~
| m
fish.cpp:33:9: error: 'odd' was not declared in this scope
33 | if(!odd) {
| ^~~
fish.cpp:34:16: error: 'all' was not declared in this scope; did you mean 'std::filesystem::perms::all'?
34 | return all;
| ^~~
| std::filesystem::perms::all
In file included from /usr/include/c++/10/filesystem:44,
from /usr/include/x86_64-linux-gnu/c++/10/bits/stdc++.h:129,
from fish.cpp:7:
/usr/include/c++/10/bits/fs_fwd.h:148:7: note: 'std::filesystem::perms::all' declared here
148 | all = 0777,
| ^~~
fish.cpp:41:8: error: 'mx' was not declared in this scope; did you mean 'm'?
41 | if(mx<=1) {
| ^~
| m
fish.cpp:46:34: warning: unused variable 'best' [-Wunused-variable]
46 | ll res = cnt2[1][n], best = res;
| ^~~~
fish.cpp:54:9: warning: unused variable 'odd' [-Wunused-variable]
54 | int odd = 0, mx = 0;
| ^~~
fish.cpp:54:18: warning: unused variable 'mx' [-Wunused-variable]
54 | int odd = 0, mx = 0;
| ^~
fish.cpp:55:8: warning: unused variable 'all' [-Wunused-variable]
55 | ll all = 0;
| ^~~
