#include "fish.h"
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define F first
#define S second
#define all(v) v.begin(),v.end()
#define sz(s) (int)s.size()
#define lb lower_bound
#define ub upper_bound
const int MAX=2e5+10;
vector<pair<ll,ll>> y[MAX];
vector<ll> p[MAX];
vector<vector<ll>> dp[MAX];
vector<ll> can[MAX];
ll getSum(int x,ll l,ll r){
if(l>r)return 0;
int L=lb(all(y[x]),make_pair(l,0ll))-y[x].begin();
int R=ub(all(y[x]),make_pair(r+1,0ll))-y[x].begin()-1;
if(L<=R){
if(L==0)return p[x][R];
return p[x][R]-p[x][L-1];
}
return 0;
}
long long max_weights(int N, int M, vector<int> X, vector<int> Y,vector<int> W) {
for(int i=0;i<M;i++){
X[i]++;
Y[i]++;
y[X[i]].pb({Y[i],i});
}
for(int i=1;i<=N;i++){
if(y[i].empty())continue;
sort(all(y[i]));
p[i].pb(W[y[i][0].S]);
for(int j=1;j<sz(y[i]);j++){
p[i].pb(p[i].back()+W[y[i][j].S]);
}
}
can[0].pb(0);
for(int i=1;i<=N;i++){
can[i].pb(0);
if(i>0){
for(auto [x,b]:y[i-1])can[i].pb(x);
}
if(i+1<N){
for(auto [x,b]:y[i+1])can[i].pb(x);
}
sort(all(can[i]));
}
for(int i=1;i<=N;i++){
dp[i].resize(sz(can[i]));
for(int j=0;j<sz(dp[i]);j++){
dp[i][j].resize(sz(can[i-1]));
for(int k=0;k<sz(can[i-1]);k++){
dp[i][j][k]=0;
}
}
}
for(int i=1;i<N;i++){
for(int j=0;j<sz(can[i]);j++){
for(int k=0;k<sz(can[i-1]);k++){
for(int f=0;f<sz(can[i+1]);f++){
ll nxt=dp[i][j][k];
{
int mx=max(can[i+1][f],can[i-1][k]);
nxt+=getSum(i,can[i][j]+1,mx);
}
dp[i+1][f][j]=max(dp[i+1][f][j],nxt);
}
}
}
}
ll ans=0;
for(int j=0;j<sz(can[N]);j++){
for(int k=0;k<sz(can[N-1]);k++){
ans=max(ans,dp[N][j][k]);
if(can[N-1][k]>=can[N][j]){
ans=max(ans,dp[N][j][k]+getSum(N,can[N][j]+1,can[N-1][k]));
}
}
}
return ans;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
65 ms |
38084 KB |
Output is correct |
2 |
Correct |
78 ms |
41392 KB |
Output is correct |
3 |
Correct |
23 ms |
28504 KB |
Output is correct |
4 |
Correct |
21 ms |
28500 KB |
Output is correct |
5 |
Execution timed out |
1078 ms |
88484 KB |
Time limit exceeded |
6 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
9 ms |
19032 KB |
Output is correct |
2 |
Runtime error |
994 ms |
2097152 KB |
Execution killed with signal 9 |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
24 ms |
28508 KB |
Output is correct |
2 |
Incorrect |
24 ms |
28504 KB |
1st lines differ - on the 1st token, expected: '882019', found: '0' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
9 ms |
19032 KB |
Output is correct |
2 |
Correct |
9 ms |
19036 KB |
Output is correct |
3 |
Incorrect |
9 ms |
19180 KB |
1st lines differ - on the 1st token, expected: '4044', found: '2022' |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
9 ms |
19032 KB |
Output is correct |
2 |
Correct |
9 ms |
19036 KB |
Output is correct |
3 |
Incorrect |
9 ms |
19180 KB |
1st lines differ - on the 1st token, expected: '4044', found: '2022' |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
9 ms |
19032 KB |
Output is correct |
2 |
Correct |
9 ms |
19036 KB |
Output is correct |
3 |
Incorrect |
9 ms |
19180 KB |
1st lines differ - on the 1st token, expected: '4044', found: '2022' |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
24 ms |
28508 KB |
Output is correct |
2 |
Incorrect |
24 ms |
28504 KB |
1st lines differ - on the 1st token, expected: '882019', found: '0' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
65 ms |
38084 KB |
Output is correct |
2 |
Correct |
78 ms |
41392 KB |
Output is correct |
3 |
Correct |
23 ms |
28504 KB |
Output is correct |
4 |
Correct |
21 ms |
28500 KB |
Output is correct |
5 |
Execution timed out |
1078 ms |
88484 KB |
Time limit exceeded |
6 |
Halted |
0 ms |
0 KB |
- |