Submission #3773365
Source Code Expand
#include <bits/stdc++.h>
using namespace std;
#ifdef _DEBUG
#define _GLIBCXX_DEBUG
#include "dump.hpp"
#else
#define dump(...)
#endif
#define int long long
#define ll long long
#define ll1 1ll
#define ONE 1ll
#define DBG 1
#define rep(i, a, b) for (int i = (a); i < (b); i++)
#define rrep(i, a, b) for (int i = (b)-1; i >= (a); i--)
#define loop(n) rep(loop, (0), (n))
#define all(c) begin(c), end(c)
const int INF =
sizeof(int) == sizeof(long long) ? 0x3f3f3f3f3f3f3f3fLL : 0x3f3f3f3f;
const int MOD = (int)(1e9) + 7;
const double PI = acos(-1);
const double EPS = 1e-9;
#define fi first
#define se second
#define pb push_back
#define eb emplace_back
using pii = pair<int, int>;
// template<class T> ostream &operator<<(ostream &os,T &t){dump(t);return os;}
template <typename T, typename S>
istream &operator>>(istream &is, pair<T, S> &p) {
is >> p.first >> p.second;
return is;
}
template <typename T, typename S>
ostream &operator<<(ostream &os, pair<T, S> &p) {
os << p.first << " " << p.second;
return os;
}
template <typename T> void printvv(const vector<vector<T>> &v) {
cerr << endl;
rep(i, 0, v.size()) rep(j, 0, v[i].size()) {
if (typeid(v[i][j]).name() == typeid(INF).name() and v[i][j] == INF) {
cerr << "INF";
} else
cerr << v[i][j];
cerr << (j == v[i].size() - 1 ? '\n' : ' ');
}
cerr << endl;
}
/*
typedef __int128_t Int;
std::ostream &operator<<(std::ostream &dest, __int128_t value) {
std::ostream::sentry s(dest);
if (s) {
__uint128_t tmp = value < 0 ? -value : value;
char buffer[128];
char *d = std::end(buffer);
do {
--d;
*d = "0123456789"[tmp % 10];
tmp /= 10;
} while (tmp != 0);
if (value < 0) {
--d;
*d = '-';
}
int len = std::end(buffer) - d;
if (dest.rdbuf()->sputn(d, len) != len) {
dest.setstate(std::ios_base::badbit);
}
}
return dest;
}
__int128 parse(string &s) {
__int128 ret = 0;
for (int i = 0; i < s.length(); i++)
if ('0' <= s[i] && s[i] <= '9')
ret = 10 * ret + s[i] - '0';
return ret;
}
*/
#ifndef _DEBUG
#define printvv(...)
#endif
void YES(bool f) { cout << (f ? "YES" : "NO") << endl; }
void Yes(bool f) { cout << (f ? "Yes" : "No") << endl; }
template <class T> bool chmax(T &a, const T &b) {
if (a < b) {
a = b;
return true;
}
return false;
}
template <class T> bool chmin(T &a, const T &b) {
if (a > b) {
a = b;
return true;
}
return false;
}
int ans = 0;
vector<pii> g[111111];
int up[111111], down[111111];
void dfs(int p, int v) {
vector<pii> lis;
lis.eb(0, 0);
for (auto e : g[v]) {
if (e.fi == p)
continue;
dfs(v, e.fi);
if (e.se == 0) {
lis.eb(up[e.fi] + 1, -1);
} else if (e.se == 1) {
lis.eb(-1, down[e.fi] + 1);
} else {
lis.eb(up[e.fi] + 1, down[e.fi] + 1);
}
}
int ma = 0;
rep(i, 0, lis.size()) {
chmax(ans, ma + lis[i].se);
chmax(ma, lis[i].fi);
}
ma = 0;
rrep(i, 0, lis.size()) {
chmax(ans, ma + lis[i].se);
chmax(ma, lis[i].fi);
}
rep(i, 0, lis.size()) {
chmax(up[v], lis[i].fi);
chmax(down[v], lis[i].se);
}
}
signed main(signed argc, char *argv[]) {
cin.tie(0);
ios::sync_with_stdio(false);
cout << fixed << setprecision(12);
int N;
cin >> N;
loop(N - 1) {
int a, b, t;
cin >> a >> b >> t;
a--, b--;
if (t == 1) {
g[a].eb(b, 1);
g[b].eb(a, 0);
} else {
g[a].eb(b, 2);
g[b].eb(a, 2);
}
}
dfs(-1,0);
cout << ans-1 << endl;
return 0;
}
Submission Info
Submission Time |
|
Task |
D - Longest Path |
User |
norma |
Language |
C++14 (GCC 5.4.1) |
Score |
100 |
Code Size |
3649 Byte |
Status |
AC |
Exec Time |
71 ms |
Memory |
25216 KB |
Judge Result
Set Name |
All |
Score / Max Score |
100 / 100 |
Status |
|
Set Name |
Test Cases |
All |
00-sample-00.txt, 00-sample-01.txt, 10-path-00.txt, 10-path-01.txt, 10-path-02.txt, 10-path-03.txt, 10-path-04.txt, 20-path_special-00.txt, 20-path_special-01.txt, 20-path_special-02.txt, 30-star-00.txt, 30-star-01.txt, 30-star-02.txt, 30-star-03.txt, 40-binary-00.txt, 40-binary-01.txt, 40-binary-02.txt, 40-binary-03.txt, 40-binary-04.txt, 50-random-00.txt, 50-random-01.txt, 50-random-02.txt, 50-random-03.txt, 50-random-04.txt, 50-random-05.txt, 50-random-06.txt, 50-random-07.txt, 50-random-08.txt, 50-random-09.txt, 50-random-10.txt, 50-random-11.txt, 50-random-12.txt |
Case Name |
Status |
Exec Time |
Memory |
00-sample-00.txt |
AC |
2 ms |
2816 KB |
00-sample-01.txt |
AC |
2 ms |
2816 KB |
10-path-00.txt |
AC |
2 ms |
2816 KB |
10-path-01.txt |
AC |
71 ms |
25088 KB |
10-path-02.txt |
AC |
17 ms |
8576 KB |
10-path-03.txt |
AC |
23 ms |
8960 KB |
10-path-04.txt |
AC |
42 ms |
15488 KB |
20-path_special-00.txt |
AC |
67 ms |
25216 KB |
20-path_special-01.txt |
AC |
65 ms |
20864 KB |
20-path_special-02.txt |
AC |
67 ms |
19712 KB |
30-star-00.txt |
AC |
40 ms |
10608 KB |
30-star-01.txt |
AC |
3 ms |
3072 KB |
30-star-02.txt |
AC |
19 ms |
6516 KB |
30-star-03.txt |
AC |
19 ms |
6516 KB |
40-binary-00.txt |
AC |
54 ms |
10368 KB |
40-binary-01.txt |
AC |
49 ms |
9728 KB |
40-binary-02.txt |
AC |
3 ms |
3072 KB |
40-binary-03.txt |
AC |
42 ms |
8576 KB |
40-binary-04.txt |
AC |
4 ms |
3200 KB |
50-random-00.txt |
AC |
53 ms |
9600 KB |
50-random-01.txt |
AC |
20 ms |
5120 KB |
50-random-02.txt |
AC |
33 ms |
6912 KB |
50-random-03.txt |
AC |
24 ms |
5888 KB |
50-random-04.txt |
AC |
28 ms |
6272 KB |
50-random-05.txt |
AC |
52 ms |
9088 KB |
50-random-06.txt |
AC |
5 ms |
3328 KB |
50-random-07.txt |
AC |
10 ms |
3840 KB |
50-random-08.txt |
AC |
49 ms |
8832 KB |
50-random-09.txt |
AC |
35 ms |
7040 KB |
50-random-10.txt |
AC |
33 ms |
6912 KB |
50-random-11.txt |
AC |
11 ms |
3968 KB |
50-random-12.txt |
AC |
52 ms |
9216 KB |