Solution

这题的解法很妙啊... 考虑这三个点可能的形态: 令它们的重心为距离到这三个点都相同的节点, 则其中两个点分别在重心的两棵子树中, 且到重心的距离相等; 第三个点可能在重心的一棵不同于前两个点子树上, 也有可能在重心往上走可以到达的位置上.

定义数组\(f[i][j]\)表示在以\(i\)为根的子树下与\(i\)的距离为\(j\)的节点个数; \(g[i][j]\)表示在以\(i\)为根的子树下, 有多少个点对满足如下条件: 这个点对到它们LCA的距离相同, 我们假设其为\(d\), 则\(i\)到它们的LCA的距离为\(d - i\), 也就是说, 假如这两个点要找到一个在\(i\)上方的第三个点组成一组答案, 则第三个点到\(i\)的距离为\(j\).

考虑枚举每个点作为重心的情况. 进行一次DFS, 令\(u\)为当前点, \(v\)为\(u\)的一个子节点, 则有:

\[ans += \sum_i g[u][i] \times f[v][i - 1] + g[v][i] \times f[u][i - 1] \\
g[u][i] += g[v][i + 1] + f[u][i] \times f[v][i - 1] \\
f[u][i] += f[v][i - 1] \\
\]

然后我们发现这种方法的转移是\(O(n^2)\)的... 考虑如何优化: 我们注意到, 当一个点\(u\)计算其第一个子节点时, 可以直接将\(f[u][i]\)赋值为\(f[v][i - 1]\), \(g[u][i]\)赋值为\(g[v][i + 1]\), 因此在计算完这个子节点后, 直接对返回的数组指针进行位移就可以得到当前点的\(f\)和\(g\). 因此考虑采用按深度树链剖分的方法, 从重儿子处继承\(f\)和\(g\)数组.

时间复杂度: \(O(n)\). 为什么? 不会证. 以后学了长链剖分再填坑吧.

由于数组是动态开的, 同时还存在指针变化的操作, 因此边界可能比较难计算. 假如你比较懒, 就直接将数组大小/对答案贡献的范围调大一些, 这样可以省去不少麻烦.

#include <cstdio>
#include <cctype>
#include <vector>
#include <algorithm>
#include <cstring> namespace Zeonfai
{
inline int getInt()
{
int a = 0, sgn = 1;
char c;
while(! isdigit(c = getchar())) if(c == '-') sgn *= -1;
while(isdigit(c)) a = a * 10 + c - '0', c = getchar();
return a * sgn;
}
}
const int N = (int)5e4;
int n;
struct result
{
long long *first, *second;
inline result() {}
inline result(long long *_first, long long *_second)
{
first = _first; second = _second;
}
};
long long ans;
struct tree
{
struct node
{
std::vector<node*> edg;
int maxDepth, dep;
node *hvy;
inline void clear()
{
edg.clear(); hvy = NULL;
}
}nd[N + 1];
inline void clear()
{
for(int i = 1; i <= n; ++ i) nd[i].clear();
}
inline void addEdge(int u, int v)
{
nd[u].edg.push_back(nd + v); nd[v].edg.push_back(nd + u);
}
void getDepth(node *u, node *pre)
{
u->maxDepth = u->dep = pre == NULL ? 0 : pre->dep + 1;
for(auto v : u->edg) if(v != pre)
{
getDepth(v, u); u->maxDepth = std::max(u->maxDepth, v->maxDepth);
if(u->hvy == NULL || v->maxDepth > u->hvy->maxDepth) u->hvy = v;
}
}
result decomposition(node *u, node *pre, node *tp)
{
result res;
long long *f, *g;
if(u->hvy != NULL) res = decomposition(u->hvy, u, tp), f = res.first - 1, g = res.second + 1;
else
{
int len = u->dep - tp->dep + 10; //懒得想边界了, 直接开大一些, 求对答案的时候也求多一些就可以了
f = new long long[len << 1]; memset(f, 0, len << 1 << 3); f += len;
g = new long long[len << 1]; memset(g, 0, len << 1 << 3);
}
f[0] = 1; ans += g[0];
for(auto v : u->edg) if(v != pre && v != u->hvy)
{
int len = v->maxDepth - v->dep + 1;
res = decomposition(v, u, v); long long *_f = res.first, *_g = res.second;
for(int i = 1; i <= len; ++ i) ans += g[i] * _f[i - 1] + _g[i] * f[i - 1];
for(int i = 1; i <= len; ++ i) g[i] += _f[i - 1] * f[i];
for(int i = 0; i <= len; ++ i) g[i] += _g[i + 1];
for(int i = 1; i <= len; ++ i) f[i] += _f[i - 1];
}
return result(f, g);
}
}T;
int main()
{ #ifndef ONLINE_JUDGE freopen("thr.in", "r", stdin);
freopen("thr.out", "w", stdout); #endif using namespace Zeonfai;
while(n = getInt())
{
T.clear();
for(int i = 1; i < n; ++ i)
{
int u = getInt(), v = getInt();
T.addEdge(u, v);
}
T.getDepth(T.nd + 1, NULL);
ans = 0;
T.decomposition(T.nd + 1, NULL, T.nd + 1);
printf("%lld\n", ans);
}
}

BZOJ 4543 2016北京集训测试赛(二)Problem B: thr的更多相关文章

  1. BZOJ 4543 2016北京集训测试赛(二)Problem B: thr 既 长链剖分学习笔记

    Solution 这题的解法很妙啊... 考虑这三个点可能的形态: 令它们的重心为距离到这三个点都相同的节点, 则其中两个点分别在重心的两棵子树中, 且到重心的距离相等; 第三个点可能在重心的一棵不同 ...

  2. 【2016北京集训测试赛(二)】 thr (树形DP)

    Description 题解 (这可是一道很早就碰到的练习题然后我不会做不想做,没想到在Contest碰到欲哭无泪......) 题目大意是寻找三点对的个数,使得其中的三个点两两距离都为d. 问题在于 ...

  3. 2016北京集训测试赛(六)Problem B: 矩阵

    Solution 最小割. 参考BZOJ 3144切糕 在那道题的基础上将建图方法稍作变形: 我们对格子进行黑白染色, 对于两个格子之和\(\le k\)的限制, 就可以确定其中一个是白色格子, 一个 ...

  4. 【2016北京集训测试赛(十)】 Azelso (期望DP)

    Time Limit: 1000 ms   Memory Limit: 256 MB Description 题解 状态表示: 这题的状态表示有点难想...... 设$f_i$表示第$i$个事件经过之 ...

  5. 【2016北京集训测试赛(八)】 crash的数列 (思考题)

    Description 题解 题目说这是一个具有神奇特性的数列!这句话是非常有用的因为我们发现,如果套着这个数列的定义再从原数列引出一个新数列,它居然还是一样的...... 于是我们就想到了能不能用多 ...

  6. 【2016北京集训测试赛(十六)】 River (最大流)

    Description  Special Judge Hint 注意是全程不能经过两个相同的景点,并且一天的开始和结束不能用同样的交通方式. 题解 题目大意:给定两组点,每组有$n$个点,有若干条跨组 ...

  7. 【2016北京集训测试赛】river

    HINT 注意是全程不能经过两个相同的景点,并且一天的开始和结束不能用同样的交通方式. [吐槽] 嗯..看到这题的想法的话..先想到了每个点的度为2,然后就有点不知所措了 隐隐约约想到了网络流,但并没 ...

  8. 【2016北京集训测试赛】azelso

    [吐槽] 首先当然是要orzyww啦 以及orzyxq奇妙顺推很强qwq 嗯..怎么说呢虽然说之前零零散散做了一些概d的题目但是总感觉好像并没有弄得比较明白啊..(我的妈果然蒟蒻) 这题的话可以说是难 ...

  9. [2016北京集训测试赛17]crash的游戏-[组合数+斯特林数+拉格朗日插值]

    Description Solution 核心思想是把组合数当成一个奇怪的多项式,然后拉格朗日插值..:哦对了,还要用到第二类斯特林数(就是把若干个球放到若干个盒子)的一个公式: $x^{n}=\su ...

随机推荐

  1. 【bzoj3339】Rmq Problem

    [bzoj3339]Rmq Problem   Description Input Output Sample Input 7 50 2 1 0 1 3 21 32 31 43 62 7 Sample ...

  2. laravel5.2总结--数据迁移

          迁移就像是数据库中的版本控制,它让团队能够轻松的修改跟共享应用程序的数据库结构.   1 创建一个迁移 1.1 使用artisan命令make:migration来创建一个新的迁移: ph ...

  3. Java集合中的细节问题

    1)集合不保存基本数据类型,而是会把基本数据类型装箱后保存. 2)Empty和null的区别:null是不存在,Empty已经初始化了,只不过里面是空的. 3)判断集合有效性: 先判断空,再判断emp ...

  4. Careercup - Microsoft面试题 - 4639756264669184

    2014-05-12 06:42 题目链接 原题: Write your own regular expression parser for following condition: az*b can ...

  5. IIS6.0,Apache低版本,PHP CGI 解析漏洞

    IIS6.0解析漏洞 在IIS6.0下存在这样的文件"名字.asp;名字.jpg" 代表了jpg文件可以以asp脚本类型的文件执行. 根据这个解析漏洞我们可以上传这种名字类型的图片 ...

  6. Python+Selenium练习篇之1-摘取网页上全部邮箱

    前面已经介绍了Python+Selenium基础篇,通过前面几篇文章的介绍和练习,Selenium+Python的webUI自动化测试算是入门了.接下来,我计划写第二个系列:练习篇,通过一些练习,了解 ...

  7. [oldboy-django][4python面试]cookie和session比较

    session定义(知乎网上) Session的数据不是储存在客户端上的,而是储存在服务器上的:而客户端使用Cookie储存一个服务器分配的客户端会话序号(Session ID),当客户端请求服务器时 ...

  8. SpringCloud Eureka参数配置项详解

    SpringCloud Eureka参数配置项详解(转) Eureka涉及到的参数配置项数量众多,它的很多功能都是通过参数配置来实现的,了解这些参数的含义有助于我们更好的应用Eureka的各种功能,下 ...

  9. java 例子

    1. 本章学习总结 今天主要学习了三个知识点 封装 继承 多态 2. 书面作业 Q1. java HelloWorld命令中,HelloWorld这个参数是什么含义? 今天学了一个重要的命令javac ...

  10. 带接口的webservice 的发布

    1.接口:(打上service注解) package ws.qlq.a; import javax.jws.WebService; /** * 面向接口的webservice发布方式 * * */ @ ...