oj.uz

Submission #1097137

# Submission time Handle Problem Language Result Execution time Memory
1097137 2024-10-06T08:15:51 Z vivkostov Dreaming (IOI13_dreaming) C++14
0 / 100
 32 ms 16876 KB 
#include<bits/stdc++.h>
#define endl '\n'
#include"dreaming.h"
using namespace std;
struct cell
{
    int to,cost;
};
int n,m,l,a[200005],b[200005],t[200005],used[200005],lead[200005],dp[200005],sdp[200005],mid,mind1,mind2,mind3,sma;
vector<cell>v[200005];
void read()
{
    cin>>n>>m>>l;
    for(int i=0;i<m;i++)
    {
        cin>>a[i]>>b[i]>>t[i];
    }
}
void make_dp(int beg)
{
    used[beg]=1;
    cell w;
    for(int i=0;i<v[beg].size();i++)
    {
        w=v[beg][i];
        if(!used[w.to])
        {
            make_dp(w.to);
            if(dp[beg]<dp[w.to]+w.cost)
            {
                sdp[beg]=dp[beg];
                dp[beg]=dp[w.to]+w.cost;
            }
            else sdp[beg]=max(sdp[beg],dp[w.to]+w.cost);
        }
    }
}
void find_mid(int beg,int izm)
{
    used[beg]=1;
    cell w;
    int ma=0,sma=0;
    cell ind;
    for(int i=0;i<v[beg].size();i++)
    {
        w=v[beg][i];
        if(!used[w.to]&&dp[w.to]>ma)
        {
            ma=dp[w.to];
            ind=w;
        }
    }
    for(int i=0;i<v[beg].size();i++)
    {
        w=v[beg][i];
        if(!used[w.to]&&w.to!=ind.to&&dp[w.to]+w.cost>sma)
        {
            sma=dp[w.to]+w.cost;
        }
    }
    izm=max(izm,sma)+ind.cost;
    if(dp[beg]<=max(izm,ma))
    {
        mid=beg;
    }
    else
    {
        find_mid(w.to,izm);
    }
}
void make_lead(int beg,int le)
{
    used[beg]=2;
    lead[beg]=le;
    cell w;
    for(int i=0;i<v[beg].size();i++)
    {
        w=v[beg][i];
        if(used[w.to]<2)
        {
            make_lead(w.to,le);
        }
    }
}
void find_max_lead()
{
    int ma=0,ind=-1;
    for(int i=0;i<n;i++)
    {
        if(ma<dp[lead[i]])
        {
            ma=dp[lead[i]];
            ind=lead[i];
        }
    }
    mind1=ind;
    ind=-1;
    ma=0;
    for(int i=0;i<n;i++)
    {
        if(lead[i]!=mind1&&ma<dp[lead[i]])
        {
            ma=dp[lead[i]];
            ind=lead[i];
        }
    }
    mind2=ind;
    ind=-1;
    ma=0;
    for(int i=0;i<n;i++)
    {
        if(lead[i]!=mind1&&lead[i]!=mind2&&ma<dp[lead[i]])
        {
            ma=dp[lead[i]];
            ind=lead[i];
        }
    }
    mind3=ind;
}
void find_mas()
{
    for(int i=0;i<n;i++)
    {
        sma=max(sma,dp[lead[i]]+sdp[lead[i]]);
    }
}
int travelTime(int N,int M,int L,int A[],int B[],int T[])
{
    n=N;
    m=M;
    l=L;
    for(int i=0;i<m;i++)
    {
        a[i]=A[i];
        b[i]=B[i];
        t[i]=T[i];
        cell g;
        g.to=b[i];
        g.cost=t[i];
        v[a[i]].push_back(g);
        g.to=a[i];
        v[b[i]].push_back(g);
    }
    for(int i=0;i<n;i++)
    {
        if(!used[i])make_dp(i);
    }
    memset(used,0,sizeof(used));
    for(int i=0;i<n;i++)
    {
        if(!used[i])
        {
            find_mid(i,0);
            make_lead(i,mid);
            mid=0;
        }
    }
    memset(used,0,sizeof(used));
    memset(dp,0,sizeof(dp));
    memset(sdp,0,sizeof(sdp));
    for(int i=0;i<n;i++)
    {
        if(!used[lead[i]])make_dp(lead[i]);
    }
    find_max_lead();
    find_mas();
    if(mind2==-1)return max(dp[mind1],sma);
    if(mind3==-1)return max(sma,dp[mind1]+dp[mind2]+l);
    return max(max(sma,dp[mind1]+dp[mind2]+l),dp[mind2]+dp[mind3]+2*l);
}
/*int main()
{
    read();
    cout<<travelTime(n,m,l,a,b,t)<<endl;
    return 0;
}
*/

Compilation message

dreaming.cpp: In function 'void make_dp(int)':
dreaming.cpp:23:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<cell>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   23 |     for(int i=0;i<v[beg].size();i++)
      |                 ~^~~~~~~~~~~~~~
dreaming.cpp: In function 'void find_mid(int, int)':
dreaming.cpp:44:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<cell>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   44 |     for(int i=0;i<v[beg].size();i++)
      |                 ~^~~~~~~~~~~~~~
dreaming.cpp:53:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<cell>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   53 |     for(int i=0;i<v[beg].size();i++)
      |                 ~^~~~~~~~~~~~~~
dreaming.cpp: In function 'void make_lead(int, int)':
dreaming.cpp:76:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<cell>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   76 |     for(int i=0;i<v[beg].size();i++)
      |                 ~^~~~~~~~~~~~~~
dreaming.cpp: In function 'void find_mid(int, int)':
dreaming.cpp:61:21: warning: 'ind.cell::cost' may be used uninitialized in this function [-Wmaybe-uninitialized]
   61 |     izm=max(izm,sma)+ind.cost;
      |         ~~~~~~~~~~~~^~~~~~~~~
dreaming.cpp:62:14: warning: 'w.cell::to' may be used uninitialized in this function [-Wmaybe-uninitialized]
   62 |     if(dp[beg]<=max(izm,ma))
      |        ~~~~~~^
dreaming.cpp:56:23: warning: 'ind.cell::to' may be used uninitialized in this function [-Wmaybe-uninitialized]
   56 |         if(!used[w.to]&&w.to!=ind.to&&dp[w.to]+w.cost>sma)
      |            ~~~~~~~~~~~^~~~~~~~~~~~~~

Subtask #1
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 32 ms 15952 KB Output is correct
2 Correct 32 ms 16876 KB Output is correct
3 Correct 24 ms 15656 KB Output is correct
4 Correct 7 ms 9736 KB Output is correct
5 Correct 7 ms 9308 KB Output is correct
6 Correct 11 ms 10504 KB Output is correct
7 Incorrect 3 ms 8284 KB Output isn't correct
8 Halted 0 ms 0 KB -

Subtask #2
0 / 10.0

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

Subtask #3
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 32 ms 15952 KB Output is correct
2 Correct 32 ms 16876 KB Output is correct
3 Correct 24 ms 15656 KB Output is correct
4 Correct 7 ms 9736 KB Output is correct
5 Correct 7 ms 9308 KB Output is correct
6 Correct 11 ms 10504 KB Output is correct
7 Incorrect 3 ms 8284 KB Output isn't correct
8 Halted 0 ms 0 KB -

Subtask #4
0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 15 ms 10332 KB Output is correct
2 Correct 22 ms 10364 KB Output is correct
3 Correct 15 ms 10328 KB Output is correct
4 Correct 15 ms 11100 KB Output is correct
5 Correct 14 ms 11092 KB Output is correct
6 Correct 21 ms 11356 KB Output is correct
7 Correct 15 ms 11356 KB Output is correct
8 Correct 14 ms 11100 KB Output is correct
9 Correct 14 ms 11100 KB Output is correct
10 Correct 15 ms 11356 KB Output is correct
11 Correct 3 ms 8284 KB Output is correct
12 Correct 4 ms 7772 KB Output is correct
13 Correct 5 ms 8796 KB Output is correct
14 Correct 5 ms 8796 KB Output is correct
15 Correct 5 ms 8796 KB Output is correct
16 Correct 5 ms 8732 KB Output is correct
17 Correct 4 ms 8792 KB Output is correct
18 Correct 6 ms 8796 KB Output is correct
19 Correct 4 ms 8796 KB Output is correct
20 Correct 3 ms 7300 KB Output is correct
21 Incorrect 3 ms 8284 KB Output isn't correct
22 Halted 0 ms 0 KB -

Subtask #5
0 / 12.0

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

Subtask #6
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 32 ms 15952 KB Output is correct
2 Correct 32 ms 16876 KB Output is correct
3 Correct 24 ms 15656 KB Output is correct
4 Correct 7 ms 9736 KB Output is correct
5 Correct 7 ms 9308 KB Output is correct
6 Correct 11 ms 10504 KB Output is correct
7 Incorrect 3 ms 8284 KB Output isn't correct
8 Halted 0 ms 0 KB -