Submission #1099784

#	Time	Username	Problem	Language	Result	Execution time	Memory
1099784		model_code	Nile (IOI24_nile)	C++17	17 / 100	2062 ms	6700 KiB

nile

// time_limit/hazem_dp_qn2.cpp

#include "bits/stdc++.h"
#include "nile.h"
using namespace std;

const long long INF = 1e14;

vector<long long> calculate_costs(vector<int> W, vector<int> A, vector<int> B, vector<int> E) {
    int n = W.size();
    int q = E.size();

    vector<int> ordW(n);
    iota(ordW.begin() ,ordW.end() ,0);
    sort(ordW.begin() ,ordW.end() ,[&](int i ,int j){
        return W[i] < W[j];
    });
    vector<int> w(n), a(n), b(n);
    for(int i = 0; i < n; i++){
        int j = ordW[i];
        w[i] = W[j];
        a[i] = A[j];
        b[i] = B[j];
    }

    auto solve = [&](int d){
        vector<vector<long long>> dp(2, vector<long long>(n+1, INF));
        dp[n&1][n] = 0;
        for(int i = n - 1; i >= 0; i--){
            dp[i&1][n] = min(a[i] + dp[i&1^1][n], b[i] + dp[i&1^1][i]);
            for(int j = i - 1; j >= 0 && w[i] - w[j] <= d; j--)
                dp[i&1][j] = min(a[i] + dp[i&1^1][j], b[i] + dp[i&1^1][n]);
        }
        return dp[0][n];
    };

    vector<long long> R(q);
    for(int i = 0; i < q; i++)
        R[i] = solve(E[i]);
    return R;
}

Compilation message (stderr)

nile.cpp: In lambda function:
nile.cpp:30:41: warning: suggest parentheses around arithmetic in operand of '^' [-Wparentheses]
   30 |             dp[i&1][n] = min(a[i] + dp[i&1^1][n], b[i] + dp[i&1^1][i]);
      |                                        ~^~
nile.cpp:30:62: warning: suggest parentheses around arithmetic in operand of '^' [-Wparentheses]
   30 |             dp[i&1][n] = min(a[i] + dp[i&1^1][n], b[i] + dp[i&1^1][i]);
      |                                                             ~^~
nile.cpp:32:45: warning: suggest parentheses around arithmetic in operand of '^' [-Wparentheses]
   32 |                 dp[i&1][j] = min(a[i] + dp[i&1^1][j], b[i] + dp[i&1^1][n]);
      |                                            ~^~
nile.cpp:32:66: warning: suggest parentheses around arithmetic in operand of '^' [-Wparentheses]
   32 |                 dp[i&1][j] = min(a[i] + dp[i&1^1][j], b[i] + dp[i&1^1][n]);
      |                                                                 ~^~

• Subtask 0: 0 / 0
• Subtask 1: 6 / 6
• Subtask 2: 0 / 13
• Subtask 3: 0 / 17
• Subtask 4: 11 / 11
• Subtask 5: 0 / 20
• Subtask 6: 0 / 15
• Subtask 7: 0 / 18