【codeforces 791D】 Bear and Tree Jumps
【题目链接】:http://codeforces.com/contest/791/problem/D
【题意】
你可以从树上的节点一次最多走k条边。
(称为跳一次);
树为无权树;
然后问你任意两点之间的条的次数的和为多少;
【题解】
如果k=1的话;
问题就是求任意两点之间的距离的和了;
这个可以在O(N)的复杂度下搞出来;
即
枚举每一条边;
看这条边左边的点的数目和右边的点的数目分别为多少->num1和num2
num1*num2就是经过这条边的路径个数;
累加所有的边的这个值就是任意两点之间的距离了;
但是对于k>1的情况有点不同;
可以假设上面求的k=1的情况的答案为S
则
k>2的答案应该为
(S+∑f(l,k))/k;
这里f(l,k)指的是某个路径长度为l,然后让他能够被k整除还要加上几.
f(10, 3) = 2, f(11, 3) = 1, f(12, 3) = 0.
这里的∑f(l,k)
也能在O(N)*K^2的复杂度内搞出来;
具体的
设某条路径的长度为k;
然后令这条路径对k求余结果为x
比如
有200条边它们的长度对3求余的结果都是1
则这每条边都要加上2才能被3整除;
即结果加上200*2;
然后对于树上的任意两点之间的距离(x,y);
=dep[x]+dep[y]-2*dep[t];
t是它们的最近公共祖先.
然后O(K^2)枚举这条边两端的端点的深度对k的求余结果i和j
i+j-2*dep[now]就是深度%k的结果为i的节点和深度%k的结果为j的节点的距离了.
然后%k得到dis;
用k减去dis再%k就是长度%k为dis的路径,每条路径需要额外加上的距离了.
按照这个方法搞出∑f(l,k)就好;
【完整代码】
#include <bits/stdc++.h>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define LL long long
#define rep1(i,a,b) for (int i = a;i <= b;i++)
#define rep2(i,a,b) for (int i = a;i >= b;i--)
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define rei(x) scanf("%d",&x)
#define rel(x) scanf("%lld",&x)
#define ref(x) scanf("%lf",&x)
typedef pair<int, int> pii;
typedef pair<LL, LL> pll;
const int dx[9] = { 0,1,-1,0,0,-1,-1,1,1 };
const int dy[9] = { 0,0,0,-1,1,-1,1,-1,1 };
const double pi = acos(-1.0);
const int N = 20e4+100;
vector <int> G[N];
int n,k;
LL sum[N],a[N][5+2],ans;
void in()
{
rei(n), rei(k);
rep1(i, 1, n - 1)
{
int x, y;
rei(x), rei(y);
G[x].push_back(y),G[y].push_back(x);
}
}
void dfs(int x, int fa,int dep)
{
a[x][dep%k] = sum[x] = 1;
for (int y : G[x])
{
if (y == fa) continue;
dfs(y, x, dep + 1);
rep1(i,0,k-1)
rep1(j, 0, k-1)
{
int dis = ((i + j) % k - ((dep * 2) % k) + k) % k;
int t = (k - dis + k) % k;
ans += t*a[x][i] * a[y][j];
}
rep1(i, 0, k - 1)
a[x][i] += a[y][i];
sum[x] += sum[y];
ans += (n - sum[y])*sum[y];
}
}
void o()
{
cout << ans / k << endl;
}
int main()
{
//freopen("F:\\rush.txt", "r", stdin);
in();
dfs(1, 0,0);
o();
//printf("\n%.2lf sec \n", (double)clock() / CLOCKS_PER_SEC);
return 0;
}
【codeforces 791D】 Bear and Tree Jumps的更多相关文章
- 【27.91%】【codeforces 734E】Anton and Tree
time limit per test3 seconds memory limit per test256 megabytes inputstandard input outputstandard o ...
- 【32.89%】【codeforces 574D】Bear and Blocks
time limit per test1 second memory limit per test256 megabytes inputstandard input outputstandard ou ...
- 【codeforces 791C】Bear and Different Names
[题目链接]:http://codeforces.com/contest/791/problem/C [题意] 给你n-k+1个限制 要求 a[i]..a[i]+k-1里面有相同的元素,或全都不同; ...
- 【codeforces 791B】Bear and Friendship Condition
[题目链接]:http://codeforces.com/contest/791/problem/B [题意] 给你m对朋友关系; 如果x-y是朋友,y-z是朋友 要求x-z也是朋友. 问你所给的图是 ...
- 【codeforces 791A】Bear and Big Brother
[题目链接]:http://codeforces.com/contest/791/problem/A [题意] 给你两个数字a和b; a每次乘3,b每次乘2 问你什么时候a第一次大于b [题解] 傻逼 ...
- 【Codeforces 639B】Bear and Forgotten Tree 3
[链接] 我是链接,点我呀:) [题意] [题解] 首先,因为高度是h 所以肯定1下面有连续的h个点依次连成一条链.->用了h+1个点了 然后,考虑d这个约束. 会发现,形成d的这个路径,它一定 ...
- 【19.27%】【codeforces 618D】Hamiltonian Spanning Tree
time limit per test2 seconds memory limit per test256 megabytes inputstandard input outputstandard o ...
- 【27.48%】【codeforces 699D】 Fix a Tree
time limit per test2 seconds memory limit per test256 megabytes inputstandard input outputstandard o ...
- 【19.05%】【codeforces 680D】Bear and Tower of Cubes
time limit per test2 seconds memory limit per test256 megabytes inputstandard input outputstandard o ...
随机推荐
- java与javax的区别分析
Java是一种受C语言影响的编程语言.Java和Javax本质上是与Java编程语言的上下文一起使用的包.实际上Java和Javax没有区别.这只是不同的名字. Java是一种编程语言,受到C语言的影 ...
- vs2015 EF code first 问题待解决
在vs 2013 上可以成功ef 生成代码.EF power Tools 安装在vs 2015 :一般不可安装, 把扩展名改成zip,解压缩. 打开extension.vsixmanifest文件 找 ...
- 基于am3358的led跑马灯測试
#include <sys/ioctl.h> #include<stdio.h> #include <fcntl.h> #include <sys/types ...
- Excel Add-in
Excel Add-in 前言 这个系列文章应该有一阵子没有更新了,原因是一如既往的多,但是根本所在是我对于某些章节其实还没有完全想好怎么写,尤其是对于Office Add-in这块 —— 到底是要每 ...
- java调用C++的过程
转自https://blog.csdn.net/yjhdxflqm/article/details/50503551 jni是java和C.C++通信的桥梁. java适合写上层的应用,C.C++适合 ...
- decode与case when
语法 decode(条件,值1,返回值1,值2,返回值2,...值n,返回值n,缺省值) select * from reglike; ,),'aaa','yes','no') decode from ...
- LA 2678 – Subsequence
看到限时3S,自己写了一个二重循环的,然后华丽的 TLE...T T 瞄了瞄书上,作者的思路果然是很好.膜拜中. 他只枚举了终点,然后用二分查找. 用到了lower_bound函数,这个lower_b ...
- 10.4 android输入系统_框架、编写一个万能模拟输入驱动程序、reader/dispatcher线程启动过程源码分析
1. 输入系统框架 android输入系统官方文档 // 需FQhttp://source.android.com/devices/input/index.html <深入理解Android 卷 ...
- JAVA日志库2
一.常用日志Jar关系 2015第30周四Java日志组件 接口:将所有日志实现适配到了一起,用统一的接口调用. 实现:目前主流的日志实现 旧日志到slf4j的适配器:如果使用了slf4j,但是只想用 ...
- 【BZOJ 3156】防御准备
[链接] 链接 [题意] 在这里输入题意 [题解] 把a倒过来 设f[i]表示在i放一个防御塔的最小花费; 我们如果从j转移过来 就表示j+1..i-1这一段放人偶. s[i] = 1 + 2 + . ...