oj.uz

Submission #854542

# Submission time Handle Problem Language Result Execution time Memory
854542 2023-09-27T22:23:14 Z EJIC_B_KEDAX Sightseeing in Kyoto (JOI22_kyoto) C++14
0 / 100
 1 ms 348 KB 
#include <bits/stdc++.h>
#include <random>

#ifndef LOCAL
    //#pragma GCC optimize("O3")
    //#pragma GCC optimize("Ofast")
    //#pragma GCC optimize("unroll-loops")
    //#pragma GCC target("avx,avx2,bmi,bmi2,popcnt,lzcnt")
#endif
using namespace std;
using ll = long long;
using dd = double;
using ld = long double;
using uii = unsigned int;
//#define x first
//#define y second
//#define all(x) x.begin(), x.end()
//#define rall(x) x.rbegin(), x.rend()


void solve();

mt19937_64 mt(1);

int32_t main() {
#ifdef LOCAL
    freopen("input.in", "r", stdin);
    freopen("output.out", "w", stdout);
#else
    ios::sync_with_stdio(false);
    cin.tie(nullptr); cout.tie(nullptr);
    //freopen("amusing.in", "r", stdin);
    //freopen("amusing.out", "w", stdout);
#endif
    cout << fixed << setprecision(30);
    int tests = 1;
    //cin >> tests;
    while (tests--) {
        solve();
    }
}

void solve() {
    int n, m;
    cin >> n >> m;
    vector<ll> a(n), b(m);
    vector<pair<ll, int>> bs(m);
    set<int> indx;
    for (int i = 0; i < n; i++) {
        cin >> a[i];
    }
    for (int i = 0; i < m; i++) {
        cin >> b[i];
        bs[i].first = b[i];
        bs[i].second = i;
        if (i < 100 || m - i < 101) {
            indx.insert(i);
        }
    }
    sort(b.begin(), b.end());
    for (int i = 0; i < min(100, m); i++) {
        indx.insert(bs[i].second);
    }
    for (int i = 0; i < m; i++) {
        bs[i].second *= -1;
    }
    sort(b.begin(), b.end());
    for (int i = 0; i < min(100, m); i++) {
        indx.insert(bs[i].second);
    }
    vector<int> ch;
    for (int i : indx) {
        ch.push_back(i);
    }
    vector<ll> dp(n, INT64_MAX), cnt(m, n - 1);
    dp[n - 1] = a[n - 1] * (m - 1);
    for (int i = n - 2; i >= 0; i--) {
        int ind = -1;
        for (int j : ch) {
            ll c = dp[cnt[j]] + b[j] * (cnt[j] - i) - a[cnt[j]] * j + a[i] * j;
            if (dp[i] > c) {
                dp[i] = c;
                ind = j;
            }
        }
        for (int j = 0; j <= ind; j++) {
            cnt[j] = i;
        }
    }
    cout << dp[0] << '\n';
}

Subtask #1
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Incorrect 0 ms 348 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #2
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Incorrect 1 ms 344 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #3
0 / 60.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Incorrect 0 ms 348 KB Output isn't correct
3 Halted 0 ms 0 KB -