test20181016 B君的第三题
题意
B 君的第三题(haskell)
题目描述
大学四年,我为什么,为什么不好好读书,没找到和你一样的工作。
B 君某天看到了这样一个题,勾起了无穷的回忆。
输入\(n, k\) 和一棵\(n\) 个点的树,有边权,没有点权。两点\(i, j\) 之间的距离\(D(i, j)\) 定义为路径上的边权和。求
\]
换句话说,枚举无序的两个点,求出距离除以\(k\) 上取整的和。
输入格式
输入第一行包含两个整数\(n, k\)。
接下来\(n-1\) 行,每行三个整数\(x, y, z\),表示\(x\) 和\(y\) 之间有一条边,边权为\(z\)。
输出格式
输出一行一个整数,表示答案。
样例输入
4 6
1 2 2
1 3 3
1 4 4
样例输出
7
数据规模与约定
对于100% 的数据,满足\(1 \leq n \leq 100000, 1 \leq k \leq 10\)。
对于100% 的数据,满足\(1 \leq x, y \leq n, 1 \leq z \leq 10\)。
对于30% 的数据,满足\(n \leq 1000\)。
对于另20% 的数据,满足\(k = 1\)。
对于另20% 的数据,满足\(k = 2\)。
分析
先考虑没有取整的情况,是一个经典题。
然后考虑有多少条路径模 k 余 1, 2, . . .。
这题是NOI2011道路修建和BZOJ2152聪聪可可改编而成的一道好题。
\]
所以我们需要求出分别有多少路径模k等于1,2,..k-1。
考虑dp。
用\(f(i,j)\)表示i的子树中有多少点到i的距离mod k = j,\(c(i)\)表示全局有多少路径mod k = i。
f、c都能在dfs的时候求出,注意顺序,不能用子树乘自己本身的方案。
代码
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#include<string>
#include<vector>
#include<list>
#include<deque>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<bitset>
#include<algorithm>
#include<complex>
#define rg register
#define il inline
#define co const
#pragma GCC optimize ("O0")
using namespace std;
template<class T> il T read(rg T&x)
{
rg T data=0;
rg int w=1;
rg char ch=getchar();
while(!isdigit(ch))
{
if(ch=='-')
w=-1;
ch=getchar();
}
while(isdigit(ch))
data=10*data+ch-'0',ch=getchar();
return x=data*w;
}
typedef long long ll;
const int INF=0x7fffffff;
const int MAXN=1e5+7,MAXK=11;
int n,k;
struct Edge
{
int nx,to,w;
}E[MAXN<<1];
int head[MAXN],ecnt;
void addedge(rg int x,rg int y,rg int w)
{
E[++ecnt].to=y,E[ecnt].w=w;
E[ecnt].nx=head[x],head[x]=ecnt;
}
int siz[MAXN];
int f[MAXN][MAXK];
ll c[MAXN];
ll ans;
il void dfs(rg int x,rg int fa)
{
siz[x]=1;
f[x][0]=1;
for(rg int i=head[x];i;i=E[i].nx)
{
rg int y=E[i].to,w=E[i].w;
if(y==fa)
continue;
dfs(y,x);
siz[x]+=siz[y];
ans += (ll) (n - siz[y]) * siz[y] * w;
for(rg int p=0;p<k;++p)
for(rg int q=0;q<k;++q)
c[(p + q + w) % k] += f[x][p] * f[y][q];
for(rg int j=0;j<k;++j)
f[x][(j + w) % k] += f[y][j];
}
}
int main()
{
freopen("haskell.in","r",stdin);
freopen("haskell.out","w",stdout);
read(n);read(k);
for(rg int i=1;i<n;++i)
{
rg int x,y,w;
read(x);read(y);read(w);
addedge(x,y,w);
addedge(y,x,w);
}
dfs(1,0);
for(rg int i=1;i<k;++i)
{
ans += (k-i) * c[i];
}
printf("%lld\n",ans/k);
// fclose(stdin);
// fclose(stdout);
return 0;
}
test20181016 B君的第三题的更多相关文章
- test20181015 B 君的第三题
题意 B 君的第三题(zhengzhou) 题目描述 让你在战争和耻辱中做一块选择,你选择耻辱,可你将来还得进行战争. 在平面上有n 个整点(横纵坐标都是整数) B 君想找到一个整点,使得这个点,到所 ...
- test20181018 B君的第三题
题意 B 君的第三题(shenyang) 题目描述 客似云来,万里无云 B 君得到了一个数组\(\{a_1,a_2,\dots,a_n\}\). B 君想通过修改让数组中个每对数都互质. 每次使一个数 ...
- test20181019 B君的第三题
题意 B 君的第三题(urumqi) 题目描述 风雨如晦,鸡鸣不已. B 君最近在研究自己的学长都在做什么工作,每个学长属于一个公司. B 君会获得一些信息,比如x 和y 在相同公司,x 和y 在不同 ...
- noiac132 B君的第三题 (树形dp)
传送门 本来想用点分治做,结果root又求不对 算的时候还算错了 我好菜啊 结果szr大佬告诉我是树形dp 我好菜啊!! 我们有$\lceil \frac{x}{k} \rceil = \frac{x ...
- shanquan2的两年三题系列
好像只有2个月就退役啦 不管了,先说一下哪三题:多点求值.lcm.替罪羊树(bzoj3065) [upd0]2016.3.29 多点求值A掉啦,myy卡常数sxbk(不是说好的是shanquan2出的 ...
- Java-集合-第三题 有如下Student 对象, private String name; private int age; private int score; private String classNum; 其中,classNum 表示学生的班号,例如“class05”。 有如下List List list = new ArrayList(); l
第三题 有如下Student 对象, private String name; private int age; private int score; private String classNum; ...
- NOIP 2008提高组第三题题解by rLq
啊啊啊啊啊啊今天已经星期三了吗 那么,来一波题解吧 本题地址http://www.luogu.org/problem/show?pid=1006 传纸条 题目描述 小渊和小轩是好朋友也是同班同学,他们 ...
- NOIP2008提高组(前三题) -SilverN
此处为前三题,第四题将单独发布 火柴棒等式 题目描述 给你n根火柴棍,你可以拼出多少个形如“A+B=C”的等式?等式中的A.B.C是用火柴棍拼出的整数(若该数非零,则最高位不能是0).用火柴棍拼数字0 ...
- 《学习OpenCV》练习题第四章第三题b
#include <highgui.h> #include <cv.h> #include "opencv_libs.h" /* *<学习OpenCV ...
随机推荐
- 2017-2018 ACM-ICPC German Collegiate Programming Contest (GCPC 2017) Solution
A. Drawing Borders Unsolved. B. Buildings Unsolved. C. Joyride Upsolved. 题意: 在游乐园中,有n个游玩设施,有些设施之间有道路 ...
- ListView的ScrollBar设置
默认ListView的滑动时,右侧会有滑动条显示,等ListView滑动结束时,滑动条消失.修改ScrollBar的显示可以在XML以及CODE中实现. CODE中实现:1.setFastScroll ...
- Ubuntu16.04桌面系统如何配置和启动wireshark
上一篇介绍了在Ubuntu系统中安装wireshark 本篇介绍在Ubuntu系统中配置和启动wireshark: 安装好后,直接在终端运行$ wireshark.出于安全方面的考虑,普通用户不能够打 ...
- uboot能ping通本机无法ping通本机上搭建的虚拟机
注意 转载请注明出处:https://www.cnblogs.com/dakewei 一.背景 1.1 uboot不能被其它主机ping通,这是由于uboot没有对其它主机发送过来的arp包进行响应, ...
- OpenDayLight Helium安装
参照:OpenDaylight的Helium(氦)版本安装 下载链接地址为 http://www.opendaylight.org/software/downloads/helium 安装: unzi ...
- Tensorboard 的使用笔记
参考的教程: https://www.tensorflow.org/guide/summaries_and_tensorboard 遇到的错误: File "/usr/local/lib/p ...
- [html5]HTML5中<section>和<article>的区别
一.section元素 从字面理解就是区块.部分的意思,相对于article元素更加广泛,每个区块都可以使用,比如页面里的导航菜单.文章正文.文章的评论等. 1.section元素用于对网站或应用程序 ...
- UVA-11925 Generating Permutations (逆向思维)
题目大意:给出1~n的某个排列,问由升序变到这个排列最少需要几次操作.操作1:将头两个数交换:操作2:将头一个数移动最后一个位置. 题目分析:反过来考虑,将这个排列变为升序排列,那么这个变化过程实际上 ...
- IOS-网络(JSON解析数据与XML解析数据)
一.JSON解析数据 // // VideoModel.h // IOS_0130_网络视频 // // Created by ma c on 16/1/30. // Copyright © 2016 ...
- JavaScript---详解scroll
scroll scroll--译为‘滚动’,他是非常常用的属性. 滚动宽高 scrollHeight scrollHeight表示元素的总高度,包括由于溢出而无法展示在网页的不可见部分(不要误解为只有 ...