Submission #37740

#	Time	Username	Problem	Language	Result	Execution time	Memory
37740		DenXman111	Hard route (IZhO17_road)	C++14	19 / 100	2000 ms	156264 KiB

road

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <map>
#include <vector>
#include <cstring>
#include <iomanip>
#include <set>
#include <sstream>
#include <ctime>

#define rust(a, b, c, d) sqrt(sqr(a - c) + sqr(b - d))
#define sqr(a) (a)*(a)
#define p_b push_back
#define m_p make_pair
#define fi first
#define se second
#define endl "\n"

typedef long long ll;
typedef long double ld;
const int MAXINT = 2147483640;
const ll MAXLL = 9223372036854775800LL;
const ll MAXN = 1123456;
const ll MOD = 1123456769;
using namespace std;

vector < int > ss[MAXN];
vector < pair < int, int > > ways[MAXN];
int len[MAXN];
void dfs(int v, int pr){
    multiset < int > s;
    ways[v].clear();
    len[v] = 0;
    for (int i = 0; i < ss[v].size(); i++) if (ss[v][i] != pr) {
        dfs(ss[v][i], v);
        len[v] = max(len[v], len[ss[v][i]] + 1);
        s.insert(len[ss[v][i]] + 1);
    }

    for (int i = 0; i < ss[v].size(); i++) if (ss[v][i] != pr) {
        set < int > :: iterator it = --(s.end());
        int x = ss[v][i];
        if (len[x] + 1 == *it) --it;
        int y = 0;
        if (s.size() > 1) y = *it;
        for (int j = 0; j < ways[x].size(); j++) ways[v].p_b(m_p(ways[x][j].fi + 1, max(ways[x][j].se, y)));
    }
    if (ss[v].size() == 1 && pr != 0) ways[v].p_b(m_p(0, 0)), len[v] = 0;
}
int main()
{
//    freopen("road.in", "r", stdin);
//    freopen("road.out", "w", stdout);
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    srand(time(0));
    int n, m, i, j, x, y;
    cin >> n;
    for (i = 1; i < n; i++){
        cin >> x >> y;
        ss[x].p_b(y);
        ss[y].p_b(x);
    }
    vector < ll > ans;
    for (i = 1; i <= n; i++) if (ss[i].size() == 1) {
        dfs(i, 0);
        for (j = 0; j < ways[i].size(); j++) ans.p_b(ll(ways[i][j].fi) * ways[i][j].se);
    }
    sort(ans.begin(), ans.end());
    if (ans.size()) x = ans.back(); else x = 0;
    y = 0;
    for (i = ans.size() - 1; i >= 0; i--) if (ans[i] == x) ++y;
    cout << x << " " << y / 2 << endl;
    return 0;
}

Compilation message (stderr)

road.cpp: In function 'void dfs(int, int)':
road.cpp:39:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int i = 0; i < ss[v].size(); i++) if (ss[v][i] != pr) {
                       ^
road.cpp:45:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int i = 0; i < ss[v].size(); i++) if (ss[v][i] != pr) {
                       ^
road.cpp:51:27: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for (int j = 0; j < ways[x].size(); j++) ways[v].p_b(m_p(ways[x][j].fi + 1, max(ways[x][j].se, y)));
                           ^
road.cpp: In function 'int main()':
road.cpp:72:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for (j = 0; j < ways[i].size(); j++) ans.p_b(ll(ways[i][j].fi) * ways[i][j].se);
                       ^
road.cpp:62:12: warning: unused variable 'm' [-Wunused-variable]
     int n, m, i, j, x, y;
            ^

• Subtask 1: 19 / 19
• Subtask 2: 0 / 33
• Subtask 3: 0 / 48