/// In The Name Of God
#pragma GCC optimize("Ofast")
#pragma GCC target_("sse,sse2,sse3,sse3,sse4,popcnt,abm,mmx")
#include <bits/stdc++.h>
#define f first
#define s second
#define pb push_back
#define pp pop_back
#define mp make_pair
#define sz(x) (int)x.size()
#define sqr(x) ((x) * 1ll * (x))
#define all(x) x.begin(), x.end()
#define Kazakhstan ios_base :: sync_with_stdio(0), cin.tie(0), cout.tie(0);
#define nl '\n'
#define ioi exit(0);
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
const int N = (int)2e5 + 7, inf = (int)1e9 + 7, mod = (int)1e9 + 7;
const ll linf = (ll)1e18 + 7;
const int dx[] = {-1, 0, 1, 0, 1, -1, -1, 1}, dy[] = {0, 1, 0, -1, 1, -1, 1, -1};
using namespace std;
int get__int() {
char x = getchar();
int res = 0;
while (!isdigit(x)) res = res * 10 + x - '0', x = getchar();
while (isdigit(x)) res = res * 10 + x - '0', x = getchar();
return res;
}
int n;
vector <int> g[N];
vector <int> cycle;
int from[N];
bool was[N], bad[N];
void find_cycle(int v = 1, int p = -1) {
was[v] = 1;
from[v] = p;
for (auto to : g[v]) {
if (to == p) continue;
if (was[to] && !sz(cycle)) {
int x = v;
while (1) {
cycle.pb(x);
bad[x] = 1;
if (x == to) break;
x = from[x];
}
}
else if (!was[to]) find_cycle(to, v);
}
}
ll ans;
int mx;
int dp[N];
void dfs(int v, int p = -1) {
priority_queue <int> st;
for (auto to : g[v]) {
if (to == p || bad[to]) continue;
dfs(to, v);
st.push(dp[to] + 1);
dp[v] = max(dp[v], dp[to] + 1);
}
int cur = 0;
for (int i = 1, j = sz(st); i <= min(2, j); i++) {
cur += st.top();
st.pop();
}
mx = max(mx, cur);
}
pair <int, int> mxv[N];
pair <int, int> dfs1(int v, int st, int p = - 1) {
map <int, int> cnt;
bool ok = 0;
for (auto to : g[v]) {
if (to == p || bad[to]) continue;
pair <int, int> res = dfs1(to, st, v);
cnt[res.f + 1] += res.s;
ok = 1;
}
mxv[v].s = 1;
if (v == st && sz(cnt)) {
auto it = --cnt.end();
mxv[v].s = it -> s;
if (it -> s > 1) {
if (2 * it -> f == mx) ans += (ll)it -> s * (it -> s - 1) / 2;
}
else if (sz(cnt) > 1) {
pair <int, int> l1 = *it, l2 = *--it;
if (l1.f + l2.f == mx) ans += (ll)l1.s * l2.s;
}
}
if (ok) {
return *--cnt.end();
}
else return {0, 1};
}
int main() {
#ifdef IOI2018
freopen ("in.txt", "r", stdin);
freopen ("C.out", "w", stdout);
#endif
Kazakhstan
cin >> n;
for (int i = 1; i <= n; i++) {
int v, u;
cin >> v >> u;
g[v].pb(u);
g[u].pb(v);
}
find_cycle();
int sz = sz(cycle);
for (auto it : cycle) {
bool ok = 0;
for (auto to : g[it]) {
if (!bad[to]) {
ok = 1;
break;
}
}
if (ok) {
dfs(it);
mx = max(mx, dp[it] + sz / 2);
}
}
for (int i = 0; i < sz; i++) {
for (int j = i + 1; j < sz; j++) {
mx = max(mx, dp[cycle[i]] + dp[cycle[j]] + min(i + (sz - j), j - i));
}
}
for (int i = 0; i < sz(cycle); i++) {
dfs1(cycle[i], cycle[i]);
}
for (int i = 0; i < sz; i++) {
for (int j = i + 1; j < sz; j++) {
int x = dp[cycle[i]] + dp[cycle[j]] + min(i + (sz - j), j - i);
if (x == mx) {
int x = mxv[cycle[i]].s * 1ll * mxv[cycle[j]].s;
ans += x;
if (i + (sz - j) == (j - i)) ans += x;
}
}
}
//cerr << ans;
cout << ans;
ioi
}
Compilation message
shymbulak.cpp:4:0: warning: ignoring #pragma GCC target_ [-Wunknown-pragmas]
#pragma GCC target_("sse,sse2,sse3,sse3,sse4,popcnt,abm,mmx")
^
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
0 ms |
10388 KB |
Output is correct |
| 2 |
Correct |
0 ms |
10388 KB |
Output is correct |
| 3 |
Correct |
0 ms |
10388 KB |
Output is correct |
| 4 |
Correct |
3 ms |
10388 KB |
Output is correct |
| 5 |
Incorrect |
0 ms |
10388 KB |
Output isn't correct |
| 6 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
3 ms |
10388 KB |
Output is correct |
| 2 |
Correct |
0 ms |
10388 KB |
Output is correct |
| 3 |
Correct |
3 ms |
10388 KB |
Output is correct |
| 4 |
Incorrect |
6 ms |
10388 KB |
Output isn't correct |
| 5 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
169 ms |
13292 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
