Submission #153709

#	Time	Username	Problem	Language	Result	Execution time	Memory
153709		Ruxandra985	Dreaming (IOI13_dreaming)	C++14	14 / 100	77 ms	12536 KiB

dreaming

#include <cstdio>
#include <vector>
#include "dreaming.h"
#include <algorithm>
#define DIMN 100010
using namespace std;
int f[DIMN] , vc , rc,dist[DIMN] , a[DIMN] , b[DIMN] , c[DIMN];
vector <pair <int,int> > v[DIMN];
pair <int,int> arb[DIMN];
int dfs_initial (int nod , int ok){ /// return diametru arbore + calcul distante
    int i,vecin , maxi1, maxi2;
    f[nod] = 1;
    for (i=0;i<v[nod].size();i++){
        vecin = v[nod][i].first;
        if (!f[vecin]){
            dfs_initial (vecin , 0);
            dist[nod] = max (dist[nod] , dist[vecin] + v[nod][i].second);
        }
    }
    if (ok == 1){ /// acum aflam diametrul
        if (v[nod].size() == 1){
            return dist[nod];
        }
        maxi1 = maxi2 = 0;
        for (i=0;i<v[nod].size();i++){
            vecin = v[nod][i].first;
            if (maxi1 < dist[vecin] + v[nod][i].second){
                maxi2 = maxi1;
                maxi1 = dist[vecin] + v[nod][i].second;
            }
            else if (maxi2 < dist[vecin] + v[nod][i].second){
                maxi2 = dist[vecin] + v[nod][i].second;
            }
        }
        return maxi1 + maxi2; /// diametrul arborelui curent
    }
    else return 0;
}
void dfs_process (int nod, int tt, int dup){
    int i,vecin , maxi1 , maxi2 , pmaxi1;
    /// daca il faci pe nod radacina , dmax e fie dist[nod] fie dup
    if (vc > max (dist[nod] , dup)){
        vc = max(dist[nod] , dup);
        rc = nod;
    }
    maxi1 = maxi2 = 0;
    pmaxi1 = 0;
    for (i=0;i<v[nod].size();i++){
        vecin = v[nod][i].first;
        if (vecin == tt)
            continue;
        if (maxi1 < dist[vecin] + v[nod][i].second){
            maxi2 = maxi1;
            maxi1 = dist[vecin] + v[nod][i].second;
            pmaxi1 = vecin;
        }
        else if (maxi2 < dist[vecin] + v[nod][i].second){
            maxi2 = dist[vecin] + v[nod][i].second;
        }
    }
    for (i=0;i<v[nod].size();i++){
        vecin = v[nod][i].first;
        if (vecin==tt)
            continue;
        if (vecin == pmaxi1)
            dfs_process (vecin , nod , v[nod][i].second + max (dup , maxi2));
        else
            dfs_process (vecin , nod , v[nod][i].second + max (dup , maxi1));
    }

}
int travelTime (int n , int m , int l , int a[] , int b[] , int c[]){
    int i , elem , inter , diam_max;
    elem = 0;
    for (i=1;i<=m;i++){
        a[i-1]++;
        b[i-1]++;
        v[a[i-1]].push_back(make_pair(b[i-1] , c[i-1]));
        v[b[i-1]].push_back(make_pair(a[i-1] , c[i-1]));
    }
    for (i=1;i<=n;i++){
        if (!f[i]){
            diam_max = max (diam_max , dfs_initial(i,1));
            vc = 2000000000;
            dfs_process (i, 0 , 0);
            /// vc rc = valoarea arborelui curent + radacina fixata
            arb[++elem] = make_pair(vc,rc);
        }
    }
    sort (arb + 1, arb + elem + 1); /// le uneste in ord descresc a vc
    inter = 0;
    for (i=elem - 1 ; i; i--){
        inter = max (inter , (elem - i) * l + arb[elem].first + arb[i].first);
    }
    return max(diam_max , inter);
}

Compilation message (stderr)

dreaming.cpp: In function 'int dfs_initial(int, int)':
dreaming.cpp:13:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (i=0;i<v[nod].size();i++){
              ~^~~~~~~~~~~~~~
dreaming.cpp:25:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for (i=0;i<v[nod].size();i++){
                  ~^~~~~~~~~~~~~~
dreaming.cpp: In function 'void dfs_process(int, int, int)':
dreaming.cpp:48:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (i=0;i<v[nod].size();i++){
              ~^~~~~~~~~~~~~~
dreaming.cpp:61:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (i=0;i<v[nod].size();i++){
              ~^~~~~~~~~~~~~~
dreaming.cpp: In function 'int travelTime(int, int, int, int*, int*, int*)':
dreaming.cpp:73:28: warning: 'diam_max' may be used uninitialized in this function [-Wmaybe-uninitialized]
     int i , elem , inter , diam_max;
                            ^~~~~~~~

• Subtask 1: 14 / 14
• Subtask 2: 0 / 10
• Subtask 3: 0 / 23
• Subtask 4: 0 / 18
• Subtask 5: 0 / 12
• Subtask 6: 0 / 23