oj.uz

Submission #1064863

# Submission time Handle Problem Language Result Execution time Memory
1064863 2024-08-18T18:37:34 Z alexdd Train (APIO24_train) C++17
0 / 100
 59 ms 14048 KB 
#include "train.h"
#include <bits/stdc++.h>
using namespace std;
const long long INF = 1e18;
struct train
{
    int from,to;
    int tin,tout;
    int cost;
};
bool cmp_trains(train x, train y)
{
    return x.tin < y.tin;
}
bool cmp_meals(pair<int,int> x, pair<int,int> y)
{
    return x.second < y.second;
}
vector<train> trains;
vector<pair<int,int>> meals;
long long dp[100005];
vector<int> meal_cost;
vector<int> train_ends[100005];

void solve_small(int i)
{
    for(int j:train_ends[trains[i].from])
    {
        if(trains[j].tout <= trains[i].tin)
        {
            long long cnt=0;
            for(auto [x,y]:meals)
                if(x > trains[j].tout && y < trains[i].tin)
                    cnt++;
            dp[i] = min(dp[i], dp[j] + cnt*meal_cost[trains[i].from] + trains[i].cost);
        }
    }
}
long long solve(int N, int M, int W, std::vector<int> T, std::vector<int> X, std::vector<int> Y,
                std::vector<int> A, std::vector<int> B, std::vector<int> C, std::vector<int> L,
                std::vector<int> R)
{
    meal_cost = T;
    for(int i=0;i<M;i++)
    {
        trains.push_back({X[i],Y[i],A[i],B[i],C[i]});
        train_ends[trains[i].to].push_back(i);
    }
    for(int i=0;i<W;i++)
    {
        meals.push_back({L[i],R[i]});
    }
    sort(trains.begin(),trains.end(),cmp_trains);
    sort(meals.begin(),meals.end(),cmp_meals);
    int poz_meals=0;
    for(int i=0;i<M;i++)
    {
        dp[i]=INF;
        if(trains[i].from==0)
        {
            long long cnt=0;
            for(auto [x,y]:meals)
                if(y < trains[i].tin)
                    cnt++;
            dp[i] = min(dp[i], cnt*meal_cost[trains[i].from] + trains[i].cost);
        }
        solve_small(i);
    }
    long long rez=INF;
    for(int i=0;i<M;i++)
    {
        if(trains[i].to==N-1)
        {
            long long cnt=0;
            for(auto [x,y]:meals)
                if(x > trains[i].tout)
                    cnt++;
            rez = min(rez, dp[i] + cnt*meal_cost[N-1]);
        }
    }
    if(rez==INF)
        return -1;
    else
        return rez;
}
/**

dp[i] = costul minim de a parcurge o ruta care incepe la nodul 0, a.i. ultimul tren luat sa fie i (cumparam un meal doar daca e obligatoriu)

*/

Compilation message

train.cpp: In function 'long long int solve(int, int, int, std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
train.cpp:55:9: warning: unused variable 'poz_meals' [-Wunused-variable]
   55 |     int poz_meals=0;
      |         ^~~~~~~~~

Subtask #1
0.0 / 5.0

# Verdict Execution time Memory Grader output
1 Incorrect 2 ms 2904 KB Wrong Answer.
2 Halted 0 ms 0 KB -

Subtask #2
0.0 / 5.0

# Verdict Execution time Memory Grader output
1 Incorrect 59 ms 14048 KB Wrong Answer.
2 Halted 0 ms 0 KB -

Subtask #3
0.0 / 30.0

# Verdict Execution time Memory Grader output
1 Incorrect 59 ms 14048 KB Wrong Answer.
2 Halted 0 ms 0 KB -

Subtask #4
0.0 / 60.0

# Verdict Execution time Memory Grader output
1 Incorrect 2 ms 2904 KB Wrong Answer.
2 Halted 0 ms 0 KB -