Submission #901852

# Submission time Handle Problem Language Result Execution time Memory
901852 2024-01-10T02:45:04 Z iliccmarko Catfish Farm (IOI22_fish) C++17
3 / 100
237 ms 38740 KB
#include <bits/stdc++.h>
#include <stdio.h>
using namespace std;
#define ll long long
#define endl "\n"
#define INF 1000000000
#define LINF 10000000000000000LL
#define pb push_back
#define all(x) x.begin(), x.end()
#define len(s) (int)s.size()
#define test_case { int t; cin>>t; while(t--)solve(); }
#define single_case solve();
#define line cerr<<"----------"<<endl;
#define ios { ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL); cerr.tie(NULL); }
#define mod 1000000007LL

ll levoo(int i, vector<ll> &levo, int n) {
    if (i < 0 || i >= n)
        return 0;
    return levo[i];
}

ll desnoo(int i, vector<ll> &desno, int n) {
    if (i < 0 || i >= n)
        return 0;
    return desno[i];
}

ll dpp(int i, int j, vector<vector<ll> > &dp, int n) {
    if (i < 0 || i >= n)
        return 0;
    return dp[i][j];
}

ll vall(int i, vector<ll> &val, int n) {
    if (i < 0 || i >= n)
        return 0;
    return val[i];
}

long long max_weights(int n, int m, vector<int> x, vector<int> y, vector<int> w) {
  vector<ll> val(n, 0);
  vector<set<pair<ll, ll> > > visine(n);
  vector<ll> desno(n, 0), levo(n, 0);

  for (int i = 0; i < m; ++i) {
    visine[x[i]].insert({y[i], i});
    val[x[i]] += (ll)w[i];
  }

  ll sigurno = 0;
  for (int i = 0; i < n; ++i) {
    ll cnt = 0;
    if (i - 1 >= 0) {
        for (auto u : visine[i]) {
            if (y[u.second] >= (*visine[i - 1].begin()).first)
                break;
            cnt += (ll)w[u.second];
        }
        levo[i] = cnt;
    }
    if (i + 1 < n) {
        cnt = 0;
        for (auto u : visine[i]) {
            if (y[u.second] >= (*visine[i + 1].begin()).first)
                break;
            cnt += (ll)w[u.second];
        }
        desno[i] = cnt;
    }
  }

  vector<vector<ll> > dp(n, vector<ll>(3, 0));
  for (int i = 0; i < n; ++i) {
    dp[i][0] = max({vall(i - 1, val, n) + max({dpp(i - 2, 2, dp, n), desnoo(i - 2, desno, n) + dpp(i - 2, 1, dp, n), dpp(i - 2, 0, dp, n)}), dpp(i - 1, 2, dp, n), dpp(i - 1, 0, dp, n)});
    dp[i][1] = max(dpp(i - 1, 1, dp, n), dpp(i - 1, 2, dp, n));
    dp[i][2] = ((i - 1 >= 0) ? val[i] : 0) + dpp(i - 1, 0, dp, n);
  }
  ll jos = 0;
  if (n >= 3) {
    jos = dp[n - 2][2] + levo[n - 1];
  }


  return max({dp[min(2, n - 1)][0], dp[min(2, n - 1)][2], jos});

}

Compilation message

fish.cpp: In function 'long long int max_weights(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:51:6: warning: unused variable 'sigurno' [-Wunused-variable]
   51 |   ll sigurno = 0;
      |      ^~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 55 ms 18516 KB Output is correct
2 Correct 64 ms 21332 KB Output is correct
3 Correct 11 ms 12892 KB Output is correct
4 Correct 10 ms 12824 KB Output is correct
5 Correct 237 ms 38740 KB Output is correct
6 Correct 184 ms 38740 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Incorrect 105 ms 25648 KB 1st lines differ - on the 1st token, expected: '40604614618209', found: '40479197388443'
3 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 9 ms 12892 KB Output is correct
2 Incorrect 8 ms 12892 KB 1st lines differ - on the 1st token, expected: '882019', found: '0'
3 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 348 KB 1st lines differ - on the 1st token, expected: '3', found: '2'
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 348 KB 1st lines differ - on the 1st token, expected: '3', found: '2'
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 348 KB 1st lines differ - on the 1st token, expected: '3', found: '2'
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 9 ms 12892 KB Output is correct
2 Incorrect 8 ms 12892 KB 1st lines differ - on the 1st token, expected: '882019', found: '0'
3 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 55 ms 18516 KB Output is correct
2 Correct 64 ms 21332 KB Output is correct
3 Correct 11 ms 12892 KB Output is correct
4 Correct 10 ms 12824 KB Output is correct
5 Correct 237 ms 38740 KB Output is correct
6 Correct 184 ms 38740 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Incorrect 105 ms 25648 KB 1st lines differ - on the 1st token, expected: '40604614618209', found: '40479197388443'
9 Halted 0 ms 0 KB -