【LOJ】#2587. 「APIO2018」铁人两项

题解

学习了圆方树!(其实是复习了Tarjan求点双)
我又双叒叕忘记了tarjan点双一个最重要,最重要的事情!
就是……假如low[v] >= dfn[u],我们就找到了一个点双,开始建立方点,但是,虽然这个点双包括点u,然而这个u啊,它很花心可能会在很多个点双里!首先u,不能被弹出去
其次呢,在栈里,u和这个点双其他的点,在栈里不一定是连续的一段,一般都是 u (一堆别的点) 点双里的点……,所以我们弹出到v,就结束这个点双,然后手动把u加进去

然后我们再来看这道题,我们枚举两个点,起点和终点,能当做中转点的点,一定是路上经过的点双里点的和

那么我们把所有点的值设成-1,方点的值设成点双里点的个数,然后就变成了对圆方树上所有路径统计路径长度,一个优秀做法就是枚举每个点看看会被多少条路径经过

代码

#include <iostream>
#include <cstdio>
#include <vector>
//#define ivorysi
#define pb push_back
#define MAXN 100005
#define space putchar(' ')
#define enter putchar('\n')
using namespace std;
typedef long long int64;
template<class T>
void read(T &res) {
    res = 0;char c = getchar();T f = 1;
    while(c < '0' || c > '9') {
    if(c == '-') f = -1;
    c = getchar();
    }
    while(c >= '0' && c <= '9') {
    res = res * 10 + c - '0';
    c = getchar();
    }
    res *= f;
}
template<class T>
void out(T x) {
    if(x < 0) putchar('-');
    if(x >= 10) {
    out(x / 10);
    }
    putchar('0' + x % 10);
}
struct node {
    int to,next;
}E[MAXN * 4];
int N,M,Cnt;
int head[MAXN],sumE,S;
int low[MAXN],dfn[MAXN],idx,sta[MAXN],top,tot,Val[MAXN * 4],siz[MAXN * 4];
int64 ans = 0;
vector<int> son[MAXN * 4];
void add(int u,int v) {
    E[++sumE].to = v;
    E[sumE].next = head[u];
    head[u] = sumE;
}
void Tarjan(int u,int fa) {
    dfn[u] = low[u] = ++idx;
    sta[++top] = u;
    Val[u] = -1;
    siz[u] = 1;
    for(int i = head[u] ; i ; i = E[i].next) {
    int v = E[i].to;
    if(dfn[v] && v != fa) {
        low[u] = min(dfn[v],low[u]);
    }
    else if(!dfn[v]) {
        Tarjan(v,u);
        if(low[v] >= dfn[u]) {
        ++Cnt;
        son[u].pb(Cnt);
        while(1) {
            int x = sta[top--];
            Val[Cnt]++;
            son[Cnt].pb(x);
            siz[Cnt] += siz[x];
            if(x == v) break;
        }
        Val[Cnt]++;
        siz[u] += siz[Cnt];
        }
        else low[u] = min(low[v],low[u]);
    }
    }
}
void dfs(int u) {
    if(Val[u] < 0) ans += (S - 1) * Val[u]; 
    ans += 1LL * (S - siz[u]) * siz[u] * Val[u];
    for(auto k : son[u]) {
    ans += 1LL * siz[k] * (S - siz[k]) * Val[u];
    }
    for(auto k : son[u]) dfs(k);
}
void Init() {
    read(N);read(M);
    int u,v;
    for(int i = 1 ; i <= M ; ++i) {
    read(u);read(v);
    add(u,v);add(v,u);
    }
    Cnt = N;
}
void Solve() {
    for(int i = 1 ; i <= N ; ++i) {
    if(!dfn[i]) {
        Tarjan(i,0);
        S = siz[i];
        dfs(i);
    }
    }
    printf("%lld\n",ans);
}
int main() {
#ifdef ivorysi
    freopen("f1.in","r",stdin);
#endif
    Init();
    Solve();
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/ivorysi/p/9075118.html