Submission #721754

#TimeUsernameProblemLanguageResultExecution timeMemory
721754SharkyCatfish Farm (IOI22_fish)C++17
0 / 100
93 ms9568 KiB
#include "fish.h"
#include <bits/stdc++.h>
using namespace std;

#define int long long

long long max_weights(int32_t n, int32_t m, std::vector<int32_t> x, std::vector<int32_t> y, std::vector<int32_t> w) {
    vector<int> pref(n + 1, 0), suff(n + 1, 0), s0(n + 1, 0);
    for (int i = 0; i < m; i++) {
        if (x[i] == 0) {
            pref[y[i]] += w[i];
            s0[y[i]] += w[i];
        }
        else if (x[i] == 1) suff[y[i]] += w[i];
    }
    for (int i = 1; i < n; i++) pref[i] += pref[i-1];
    for (int i = n - 2; i >= 0; i--) {
        suff[i] += suff[i+1];
        // s0[i] += s0[i + 1];
    }
    int ans = suff[0];
    for (int i = 0; i < n; i++) {
        ans = max(ans, pref[i] + suff[i + 1]);
        if (i < n - 1) ans = max(ans, suff[0] + s0[i + 1]);
    }

    return ans;
}

#ifndef EVAL

int32_t main() {
  int32_t N, M;
  assert(2 == scanf("%d %d", &N, &M));

  std::vector<int32_t> X(M), Y(M), W(M);
  for (int32_t i = 0; i < M; ++i) {
    assert(3 == scanf("%d %d %d", &X[i], &Y[i], &W[i]));
  }

  long long result = max_weights(N, M, X, Y, W);
  printf("%lld\n", result);
  return 0;
}
#endif
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...