BZOJ4543 [POI2014]Hotel加强版

Address

• BZOJ4543 BZOJ3522

Solution

• 三个点两两距离相等，相当于选出三条长度相等且只有唯一公共点的路径。
• 设 $f[x][i]$ 表示点 $x$ 的子树内与点 $x$ 距离为 $i$ 的点数，转移显然为：$$f[x][i]=\sum _{y\in children[x]}f[y][i-1]$$
• 设 $g[x][i]$ 表示在点 $x$ 的子树内选取两点，满足两点到其 $LCA$ 的距离为 $d$，$LCA$ 到点 $x$ 的距离为 $d-i$ 的方案数，转移只要额外考虑 $LCA$ 为点 $x$ （$d=i$）的情况，其余情况可以由子节点的 $g$ 得到，即：$$g[x][i]=\sum _{y\in children[x]}(g[y][i+1]+f^{\prime }[x][i]\times f[y][i-1])（f^{\prime },g^{\prime }表示处理y这棵子树前f的值）$$
• 则答案 $ans=\sum _{x=1}^n\sum _{y\in children[x]}\sum _i(f^{\prime }[x]i]\times g[y][i+1]+g^{\prime }[x][i]\times f[y][i-1])$
• 时间复杂度 $O(n^2)$。
• 考虑用类似 $dsuontree$ 的思想优化，保存长链上子节点的信息（相当于将子节点的 $f$ 右移， $g$ 左移），其余节点暴力计算，时间复杂度相当于树中所有长链的长度之和，即为 $O(n)$。
• 空间问题用指针分配即可，具体参考实现。

Code

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cctype>
#include <cmath>
#include <ctime>
 
template <class T>
inline void read(T &res)
{
    char ch; 
    while (ch = getchar(), !isdigit(ch));
    res = ch ^ 48;
    while (ch = getchar(), isdigit(ch))
        res = res * 10 + ch - 48;
}
 
typedef long long ll;
const int N = 1e5 + 5;
const int M = 1e6 + 5;
 
int n, mx[N], son[N];
ll ans, tmp[M], *f[N], *g[N], *now = tmp + N;
 
struct Edge 
{
    int to; Edge *nxt;
}p[N << 1], *lst[N], *P = p;
 
 
inline void Link(int x, int y)
{
    (++P)->nxt = lst[x]; lst[x] = P; P->to = y;
    (++P)->nxt = lst[y]; lst[y] = P; P->to = x; 
}
 
inline void Create(int x)
{
    f[x] = now;
    now += mx[x] << 1 | 1;
    g[x] = now;
    now += mx[x] << 1 | 1;
}
 
inline void Dfs1(int x, int Fa)
{
    for (Edge *e = lst[x]; e; e = e->nxt)
    {
        int y = e->to;
        if (y == Fa)    
            continue;
        Dfs1(y, x);
         
        if (mx[y] + 1 > mx[x])
        {
            mx[x] = mx[y] + 1;
            son[x] = y; 
        }
    }
}
 
inline void Dfs2(int x, int Fa)
{
    if (son[x])
    {
        f[son[x]] = f[x] + 1;
        g[son[x]] = g[x] - 1;
        Dfs2(son[x], x);
    }
    f[x][0] = 1;
    ans += g[x][0];
    for (Edge *e = lst[x]; e; e = e->nxt)
    {
        int y = e->to;
        if (y == son[x] || y == Fa)
            continue;
        Create(y);
        Dfs2(y, x);
        for (int i = 0; i <= mx[y]; ++i)
        {
            ans += g[x][i + 1] * f[y][i];
            if (i)
                ans += f[x][i - 1] * g[y][i];
        }
        for (int i = 0; i <= mx[y]; ++i)
        {
            g[x][i + 1] += f[y][i] * f[x][i + 1];
            f[x][i + 1] += f[y][i];
            if (i) 
                g[x][i - 1] += g[y][i];
        }
    }
}
 
int main()
{
    read(n);
    for (int i = 1, x, y; i < n; ++i)
    {
        read(x); read(y);
        Link(x, y);
    }
     
    Dfs1(1, 0);
    Create(1);
    Dfs2(1, 0);
    std::cout << ans << std::endl;
}