This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "fish.h"
#include <bits/stdc++.h>
using namespace std;
long long max_weights(int n,int m,vector<int>x,vector<int>y,vector<int>w){
// long long pre[n+1][n+1]{};
// for(int i = 0; i < m; i++) x[i]++,y[i]++;
// for(int i = 0; i < m; i++) pre[x[i]][y[i]] += w[i];
// for(int i = 1; i <= n; i++){
// for(int j = 1; j <= n; j++){
// pre[i][j] += pre[i][j-1];
// }
// }
long long ans = 0;
long long ans2 = 0;
int cost[n][2]{};
for(int i = 0; i < m; i++){
if(x[i] == 1){cost[y[i]][1] = w[i]; ans += w[i];}
if(x[i] == 0){ans2 += w[i];cost[y[i]][0]=w[i];}
}
if(n == 2 || n == 1) return max(ans,ans2);
ans2 = max(ans,ans2);
for(int i = 1; i < n; i++){
ans -= cost[i-1][1];
ans += cost[i-1][0];
ans2 = max(ans2,ans);
}
return ans2;
}
// int main(){
// cout << max_weights(5, 4, {0, 1, 4, 3}, {2, 1, 4, 3}, {5, 2, 1, 3}) << '\n';
// }
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |