codeforces 715c
题目大意:
给定一个有N个点的树,问其中有多少条路径满足他们的边权连成的数对M取余为0。
其中gcd(M,10)=1。
题解:
很亲民的点分治题目,对每一层点分治,预处理每个点到当前根的数字并对m取余,和当前根到每个点的数字取余。
我们得到公式:x1*10dep2+x2=0 mod(m)
dep2指终点的深度,x2指从根到终点数字,通过exgcd即可算出,若想使结果mod m=0,则出发点到根的数字必须是x1。
从公式还可以看出,若枚举出发点,我还需要知道终点的dep,不好做。
所以解法就出来了,先将所有出发点的值丢进一个map(或hash),然后枚举每个子树,先将该子树的出发点值去掉,然后根据终点的值和map计算答案,最后再将该子树出发点的值加进去。
这样可以保证计算答案的不重不漏。还有就是注意根节点的处理。
时间cf上1s2,如果map改hash,应该能更快一些。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<iomanip>
#include<map>
#include<queue>
using namespace std;
#define mem1(i,j) memset(i,j,sizeof(i))
#define mem2(i,j) memcpy(i,j,sizeof(i))
#define LL long long
#define up(i,j,n) for(int i=(j);i<=(n);++i)
#define down(i,n,j) for(int i=n;i>=j;--i)
#define Auto(i,x) for(int i=linkk[x];i;i=e[i].next)
#define FILE "dealing"
#define poi vec
#define db double
#define eps 1e-10
#define mid ((l+r)>>1)
const int maxn=101000,inf=1000000000;
int read(){
int x=0,f=1,ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0',ch=getchar();}
return f*x;
}
inline bool cmax(int& a,int b){return a<b?a=b,true:false;}
inline bool cmin(int& a,int b){return a>b?a=b,true:false;}
LL n,m,fac[maxn],ans=0;
void exgcd(LL a,LL b,LL& d,LL& x,LL& y){
if(b==0){x=1,y=0,d=a;return;}
exgcd(b,a%b,d,x,y);
LL t=x;x=y;y=t-a/b*x;
}
struct node{int y,next,v;}e[maxn<<1];int len=0,linkk[maxn<<1];
void insert(int x,int y,int v){e[++len].y=y;e[len].v=v;;e[len].next=linkk[x];linkk[x]=len;}
int siz[maxn],fa[maxn],root,q[maxn],vis[maxn],Max[maxn],Min=0,dep[maxn],head,tail;
bool b[maxn];LL w[maxn],r[maxn];
void getsize(int x){
head=tail=0;q[++tail]=x;Min=inf;
while(++head<=tail){
int x=q[head];b[x]=1;Max[x]=0;
siz[x]=1;
for(int i=linkk[x];i;i=e[i].next)
if(!b[e[i].y]&&!vis[e[i].y])fa[e[i].y]=x,q[++tail]=e[i].y;
}
down(j,tail,1){
int x=q[j];
for(int i=linkk[x];i;i=e[i].next){
if(e[i].y==fa[x]||vis[e[i].y])continue;
siz[x]+=siz[e[i].y];
cmax(Max[x],siz[e[i].y]);
}
cmax(Max[x],tail-siz[x]);
if(cmin(Min,Max[x]))root=x;//求出根节点为root
}
up(i,1,tail)b[q[i]]=0;
}
map<int,int> t;//一只桶
void getans(LL x1,LL x2,LL c){
LL d,x,y;
exgcd(x1,x2,d,x,y);
x=(x*c/d%m+m)%m;
if(t.count(x))ans+=t[x];
}
void add(int x,int num){
head=tail=0;q[++tail]=x;r[x]=w[x]=num;dep[x]=1;
while(++head<=tail){
int x=q[head];b[x]=1;
if(t.count(r[x]))t[r[x]]++;
else t[r[x]]=1;
for(int i=linkk[x];i;i=e[i].next){
if(b[e[i].y]||vis[e[i].y])continue;
dep[e[i].y]=dep[x]+1;
w[e[i].y]=(w[x]*10+e[i].v)%m;//接收端
r[e[i].y]=(r[x]+fac[dep[x]]*e[i].v)%m;//发出端
q[++tail]=e[i].y;
}
}
up(i,1,tail)b[q[i]]=0;
}
void cal(int x,int num){
head=tail=0;q[++tail]=x;r[x]=w[x]=num;dep[x]=1;
while(++head<=tail){
int x=q[head];b[x]=1;t[r[x]]--;
for(int i=linkk[x];i;i=e[i].next){
if(b[e[i].y]||vis[e[i].y])continue;
q[++tail]=e[i].y;
}
}
up(i,1,tail)b[q[i]]=0,getans(-fac[dep[q[i]]],m,w[q[i]]);
up(i,1,tail)t[r[q[i]]]++;
}
void solve(int rt){
getsize(rt);
t.clear();vis[root]=1;b[root]=1;
t[0]=1;
for(int i=linkk[root];i;i=e[i].next){
if(vis[e[i].y])continue;
add(e[i].y,e[i].v%m);
}
for(int i=linkk[root];i;i=e[i].next){
if(vis[e[i].y])continue;
cal(e[i].y,e[i].v%m);
}
ans+=t[0]-1;
for(int i=linkk[root];i;i=e[i].next){
if(vis[e[i].y])continue;
solve(e[i].y);
}
}
int main(){
freopen(FILE".in","r",stdin);
freopen(FILE".out","w",stdout);
n=read(),m=read();
fac[0]=1;up(i,1,n)fac[i]=fac[i-1]*10%m;
up(i,1,n-1){
int x=read()+1,y=read()+1,v=read();
insert(x,y,v);insert(y,x,v);
}
solve(1);
cout<<ans<<endl;
return 0;
}
codeforces 715c的更多相关文章
- [Codeforces 715C] Digit Tree
[题目链接] https://codeforces.com/contest/715/problem/C [算法] 考虑点分治 一条路径(x , y)合法当且仅当 : d(x) * 10 ^ dep(x ...
- 【Codeforces 715C】Digit Tree(点分治)
Description 程序员 ZS 有一棵树,它可以表示为 \(n\) 个顶点的无向连通图,顶点编号从 \(0\) 到 \(n-1\),它们之间有 \(n-1\) 条边.每条边上都有一个非零的数字. ...
- python爬虫学习(5) —— 扒一下codeforces题面
上一次我们拿学校的URP做了个小小的demo.... 其实我们还可以把每个学生的证件照爬下来做成一个证件照校花校草评比 另外也可以写一个物理实验自动选课... 但是出于多种原因,,还是绕开这些敏感话题 ...
- 【Codeforces 738D】Sea Battle(贪心)
http://codeforces.com/contest/738/problem/D Galya is playing one-dimensional Sea Battle on a 1 × n g ...
- 【Codeforces 738C】Road to Cinema
http://codeforces.com/contest/738/problem/C Vasya is currently at a car rental service, and he wants ...
- 【Codeforces 738A】Interview with Oleg
http://codeforces.com/contest/738/problem/A Polycarp has interviewed Oleg and has written the interv ...
- CodeForces - 662A Gambling Nim
http://codeforces.com/problemset/problem/662/A 题目大意: 给定n(n <= 500000)张卡片,每张卡片的两个面都写有数字,每个面都有0.5的概 ...
- CodeForces - 274B Zero Tree
http://codeforces.com/problemset/problem/274/B 题目大意: 给定你一颗树,每个点上有权值. 现在你每次取出这颗树的一颗子树(即点集和边集均是原图的子集的连 ...
- CodeForces - 261B Maxim and Restaurant
http://codeforces.com/problemset/problem/261/B 题目大意:给定n个数a1-an(n<=50,ai<=50),随机打乱后,记Si=a1+a2+a ...
随机推荐
- vue 权限控制按钮3种样式、内容、以及跳转事件
最近碰到一个因为要根据权限来给一个按钮变成不同功能, 简单写出3个按钮然后用v-if也能实现这个功能,但是在加载页面时,如果延迟过高则会把按钮按照DOM顺序加载出来,这是个很不好的效果 思索了下,把三 ...
- windows命令行更改IP地址
修改IP地址时,提示出现一个意外情况 netsh interface ip set address "以太网" static 192.168.3.151 255.255.255.0 ...
- jvm类加载的过程
java类加载过程:加载-->验证-->准备-->解析-->初始化,之后类就可以被使用了.绝大部分情况下是按这 样的顺序来完成类的加载全过程的.但是是有例外的地方,解析也是可以 ...
- ELK之收集haproxy日志
由于HAProxy的运行信息不写入日志文件,但它依赖于标准的系统日志协议将日志发送到远程服务器(通常位于同一系统上),所以需要借助rsyslog来收集haproxy的日志.haproxy代理nginx ...
- 一些yuv视频下载地址
因为测试需要下载一些yuv视频地址,现存一个可以下载yuv视频的地址以备后用 http://trace.eas.asu.edu/yuv/index.html ftp://ftp.ldv.e-techn ...
- PyTorch框架+Python 3面向对象编程学习笔记
一.CNN情感分类中的面向对象部分 sparse.py super(Embedding, self).__init__() 表示需要父类初始化,即要运行父类的_init_(),如果没有这个,则要自定义 ...
- Engine中如何进行七参数投影转换?
来自:http://zhihu.esrichina.com.cn/?/question/6858 解决办法]:首先创建自定义geotransformation,然后用IGeometry.Project ...
- Python爬虫之路——简单网页抓图升级版(添加多线程支持)
转载自我的博客:http://www.mylonly.com/archives/1418.html 经过两个晚上的奋斗.将上一篇文章介绍的爬虫略微改进了下(Python爬虫之路--简单网页抓图),主要 ...
- HDU 4635 Strongly connected(强连通)经典
Strongly connected Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Other ...
- Food hub
Work center List Tillage 耕作 Hand harvest 手工采收 Planting 种植 Cultivating 培养 Mulching 覆盖 Dig harvest 挖地采 ...