#include "fish.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
ll p2(int n, int m, vector<int> x, vector<int> y, vector<int> w) {
vector<int> a(n,0), b(n,0);
for (int i=0; i<m; ++i) {
if (x[i]==0) a[y[i]]=w[i];
else b[y[i]]=w[i];
}
ll ans=0;
for (int i=0; i<n; ++i) ans+=b[i];
ll cnt=ans;
for (int i=0; i<n; ++i) {
cnt+=a[i]-b[i];
ans=max(ans,cnt);
}
return ans;
}
ll p3(int n, int m, vector<int> x, vector<int> w) {
vector<int> a(n,0);
vector<pair<int,int>> b(n);
for (int i=0; i<n; ++i) b[i]={0,i};
for (int i=0; i<m; ++i) {
b[x[i]]={w[i],x[i]};
a[x[i]]=w[i];
}
sort(b.begin(), b.end());
vector<int> vis(n,0);
ll ans=0;
for (int i=0; i<n; ++i) {
auto it=b[i];
int j=it.second, w=it.first;
//for (auto x:a) cout<<x<<' '; cout<<'\n';
//for (auto x:vis) cout<<x<<' '; cout<<'\n';
//cout<<j<<' '<<w<<'\n';
if ((j?(vis[j-1]?0:a[j-1]):0) + ((j<n-1)?(vis[j+1]?0:a[j+1]):0) > a[j]) {
ans+=(j?(vis[j-1]?0:a[j-1]):0) + ((j<n-1)?(vis[j+1]?0:a[j+1]):0) - a[j];
//cout<<(j?(vis[j-1]?0:a[j-1]):0)<<' '<<((j<n-1)?(vis[j+1]?0:a[j+1]):0)<<' '<<ans<<'\n';
a[j]=0;
if (j) vis[j-1]=1;
if (j<n-1) vis[j+1]=1;
}
}
return ans;
}
ll max_weights(int n, int m, vector<int> x, vector<int> y, vector<int> w) {
int _p1=1, _p2=1, _p3=1;
ll sum=0;
for (int i=0; i<m; ++i) {
_p1&=!(x[i]&1);
_p2&=x[i]<=1;
_p3&=y[i]==0;
sum+=w[i];
}
if (_p1) return sum;
if (_p2) return p2(n,m,x,y,w);
if (_p3) return p3(n,m,x,w);
}
Compilation message
fish.cpp: In function 'll p3(int, int, std::vector<int>, std::vector<int>)':
fish.cpp:39:22: warning: unused variable 'w' [-Wunused-variable]
39 | int j=it.second, w=it.first;
| ^
fish.cpp: In function 'll max_weights(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:71:1: warning: control reaches end of non-void function [-Wreturn-type]
71 | }
| ^
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
27 ms |
2132 KB |
Output is correct |
2 |
Correct |
33 ms |
2636 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
90 ms |
7264 KB |
Output is correct |
6 |
Correct |
121 ms |
7284 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
51 ms |
6684 KB |
Output is correct |
3 |
Correct |
56 ms |
8140 KB |
Output is correct |
4 |
Correct |
22 ms |
2132 KB |
Output is correct |
5 |
Correct |
28 ms |
2628 KB |
Output is correct |
6 |
Incorrect |
1 ms |
212 KB |
1st lines differ - on the 1st token, expected: '4044', found: '6066' |
7 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
4 ms |
1876 KB |
Output is correct |
3 |
Incorrect |
25 ms |
3500 KB |
1st lines differ - on the 1st token, expected: '21261825233649', found: '16876766837917' |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
212 KB |
1st lines differ - on the 1st token, expected: '3', found: '1' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
212 KB |
1st lines differ - on the 1st token, expected: '3', found: '1' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
212 KB |
1st lines differ - on the 1st token, expected: '3', found: '1' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
4 ms |
1876 KB |
Output is correct |
3 |
Incorrect |
25 ms |
3500 KB |
1st lines differ - on the 1st token, expected: '21261825233649', found: '16876766837917' |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
27 ms |
2132 KB |
Output is correct |
2 |
Correct |
33 ms |
2636 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
90 ms |
7264 KB |
Output is correct |
6 |
Correct |
121 ms |
7284 KB |
Output is correct |
7 |
Correct |
1 ms |
212 KB |
Output is correct |
8 |
Correct |
51 ms |
6684 KB |
Output is correct |
9 |
Correct |
56 ms |
8140 KB |
Output is correct |
10 |
Correct |
22 ms |
2132 KB |
Output is correct |
11 |
Correct |
28 ms |
2628 KB |
Output is correct |
12 |
Incorrect |
1 ms |
212 KB |
1st lines differ - on the 1st token, expected: '4044', found: '6066' |
13 |
Halted |
0 ms |
0 KB |
- |