#include "fish.h"
#include <bits/stdc++.h>
#include <vector>
using namespace std;
typedef long long int ll;
long long max_weights(int N, int M, std::vector<int> X, std::vector<int> Y,std::vector<int> W) {
ll ans =0;
ll a[300005];
ll flag=0,flag1=0;
for(int i =0;i<M;i++)
{
if(X[i]%2==1)
{
flag=1;
}
if(Y[i]!=0)
{
flag1=1;
}
a[X[i]]+=W[i];
}
if(flag==0)
{
for(int i =0;i<M;i++)
{
ans+=W[i];
}
}
else if(flag1==0)
{
ll dp[300005][2];
memset(dp,0,sizeof(dp));
dp[1][0]=a[1];
dp[1][1]=a[0];
for(int i =2;i<N;i++)
{
dp[i][0]=max(dp[i-1][1]+a[i],dp[i-1][0]);
dp[i][1]=max({dp[i-2][0]+a[i-1],dp[i-1][1],dp[i-2][1]+a[i-1]});
}
ans=max(dp[N-1][0],dp[N-1][1]);
}
else
{
ll b[300005][2];
memset(b,0,sizeof(b));
for(int i =0;i<M;i++)
{
b[Y[i]][X[i]]+=W[i];
}
for(int i =1;i<N-1;i++)
{
b[i][0]+=b[i-1][0];
b[i][1]+=b[i-1][1];
}
ans=max(b[N-1][0],b[N-1][1]);
if(N==2)
{
return ans;
}
for(int i =0;i<N;i++)
{
ans=max(ans,(b[N-1][1]-b[i][1]+b[i][0]));
}
}
return ans;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
24 ms |
9192 KB |
Output is correct |
2 |
Correct |
29 ms |
9600 KB |
Output is correct |
3 |
Correct |
3 ms |
7252 KB |
Output is correct |
4 |
Correct |
3 ms |
7252 KB |
Output is correct |
5 |
Correct |
86 ms |
14304 KB |
Output is correct |
6 |
Correct |
89 ms |
14284 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
3 ms |
7252 KB |
1st lines differ - on the 1st token, expected: '2', found: '1' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
7252 KB |
Output is correct |
2 |
Correct |
4 ms |
7284 KB |
Output is correct |
3 |
Correct |
17 ms |
8532 KB |
Output is correct |
4 |
Correct |
13 ms |
8148 KB |
Output is correct |
5 |
Correct |
28 ms |
9672 KB |
Output is correct |
6 |
Correct |
23 ms |
9684 KB |
Output is correct |
7 |
Correct |
28 ms |
9668 KB |
Output is correct |
8 |
Correct |
29 ms |
9676 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
4 ms |
7252 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 |
4 ms |
7252 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 |
4 ms |
7252 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 |
3 ms |
7252 KB |
Output is correct |
2 |
Correct |
4 ms |
7284 KB |
Output is correct |
3 |
Correct |
17 ms |
8532 KB |
Output is correct |
4 |
Correct |
13 ms |
8148 KB |
Output is correct |
5 |
Correct |
28 ms |
9672 KB |
Output is correct |
6 |
Correct |
23 ms |
9684 KB |
Output is correct |
7 |
Correct |
28 ms |
9668 KB |
Output is correct |
8 |
Correct |
29 ms |
9676 KB |
Output is correct |
9 |
Incorrect |
29 ms |
9804 KB |
1st lines differ - on the 1st token, expected: '99999', found: '1' |
10 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
24 ms |
9192 KB |
Output is correct |
2 |
Correct |
29 ms |
9600 KB |
Output is correct |
3 |
Correct |
3 ms |
7252 KB |
Output is correct |
4 |
Correct |
3 ms |
7252 KB |
Output is correct |
5 |
Correct |
86 ms |
14304 KB |
Output is correct |
6 |
Correct |
89 ms |
14284 KB |
Output is correct |
7 |
Incorrect |
3 ms |
7252 KB |
1st lines differ - on the 1st token, expected: '2', found: '1' |
8 |
Halted |
0 ms |
0 KB |
- |