[JZOJ5055]树上路径
来源:互联网 发布:mac怎么将dock栏放大 编辑:程序博客网 时间:2024/03/28 18:33
题目大意
给定一棵
给定参数
注意:单点也算路径。并且路径
题目分析
很直观的想法是点剖。
考虑对于一个分治重心,我依次处理其各个子树。对于一条重心到一个点的路径,假设其权值和是
把后面的式子瞎jb化一下,变成只和
时间复杂度
还存在某排序加删掉重复部分的算法,时间复杂度相同,但是常数小很多。
代码实现
#include <iostream>#include <cstdio>#include <cctype>using namespace std;typedef long long LL;int read(){ int x=0,f=1; char ch=getchar(); while (!isdigit(ch)) f=ch=='-'?-1:f,ch=getchar(); while (isdigit(ch)) x=x*10+ch-'0',ch=getchar(); return x*f;}const int N=100050;const int E=N<<1;const int P=10000000;const int V=1000000000;const int LGV=31;const int S=N*LGV;struct segment_tree{ int son[S][2]; int size[S]; int tot; void init(){tot=0;} int newnode() { size[++tot]=0,son[tot][0]=son[tot][1]=0; return tot; } void insert(int &rt,int x,int l,int r) { if (!rt) rt=newnode(); ++size[rt]; if (l==r) return; int mid=l+r>>1; if (x<=mid) insert(son[rt][0],x,l,mid); else insert(son[rt][1],x,mid+1,r); } int query(int rt,int st,int en,int l,int r) { if (!rt) return 0; if (st==l&&en==r) return size[rt]; int mid=l+r>>1; if (en<=mid) return query(son[rt][0],st,en,l,mid); else if (mid+1<=st) return query(son[rt][1],st,en,mid+1,r); else return query(son[rt][0],st,mid,l,mid)+query(son[rt][1],mid+1,en,mid+1,r); }}t[2];int last[N],fa[N],que[N],size[N],val[N];int tmp[N][2],temp[N][2];int tov[E],nxt[E];int root[P][2];bool vis[N];int n,p,head,tail,tot,tmps,temps;LL ans;void insert(int x,int y){tov[++tot]=y,nxt[tot]=last[x],last[x]=tot;}int core(int og){ int i,x,y,ret,rets=n,tmp; for (head=0,fa[que[tail=1]=og]=0;head<tail;) for (size[x=que[++head]]=1,i=last[x];i;i=nxt[i]) if ((y=tov[i])!=fa[x]&&!vis[y]) fa[que[++tail]=y]=x; for (head=tail;head>1;--head) size[fa[que[head]]]+=size[que[head]]; for (head=1;head<=tail;++head) { for (i=last[x=que[head]],tmp=size[og]-size[x];i;i=nxt[i]) if ((y=tov[i])!=fa[x]&&!vis[y]) tmp=max(tmp,size[y]); if (rets>tmp) rets=tmp,ret=x; } return ret;}void dfs(int x,int sum,int mx){ (sum+=val[x])%=p,mx=max(mx,val[x]); ans+=t[0].query(root[(mx%p-sum+p)%p][0],0,mx,0,V); ans+=t[1].size[root[sum][1]]-t[1].query(root[sum][1],0,mx,0,V); tmp[++tmps][0]=sum,tmp[tmps][1]=mx; for (int i=last[x],y;i;i=nxt[i]) if ((y=tov[i])!=fa[x]&&!vis[y]) fa[y]=x,dfs(y,sum,mx);}void solve(int x){ int c=core(x); fa[c]=0,t[0].init(),t[1].init(); for (int i=last[c];i;i=nxt[i]) if (!vis[x=tov[i]]) { fa[x]=c,dfs(x,0,0); for (int sum,mx;tmps;--tmps) { sum=(tmp[tmps][0]+val[c]%p)%p,mx=max(tmp[tmps][1],val[c]),ans+=sum==mx%p; t[0].insert(root[sum][0],mx,0,V); t[1].insert(root[(mx%p-sum+p)%p][1],mx,0,V); temp[++temps][0]=sum,temp[temps][1]=mx; } } for (int sum,mx;temps;--temps) { sum=temp[temps][0],mx=temp[temps][1]; root[sum][0]=root[(mx%p-sum+p)%p][1]=0; } vis[c]=1; for (int i=last[c],y;i;i=nxt[i]) if (!vis[y=tov[i]]) solve(y);}int main(){ freopen("path.in","r",stdin),freopen("path.out","w",stdout); n=read(),p=read(); for (int i=1,x,y;i<n;++i) x=read(),y=read(),insert(x,y),insert(y,x); for (int i=1;i<=n;++i) val[i]=read(); solve(1); printf("%lld\n",ans+n); fclose(stdin),fclose(stdout); return 0;}
0 0
- [JZOJ5055]树上路径
- 【JZOJ5055】【GDOI2017模拟二试4.12】树上路径
- 【jzoj5055】【GDOI2017模拟二试4.12】【树上路径】【点分治】
- 树上路径
- 树上路径
- 树上路径
- noip2013truck树上路径倍增
- 树上的路径
- [bzoj3784]树上的路径
- 树上最长单色路径
- 树上最长单色路径
- 树上最长单色路径
- 3784: 树上的路径
- 【JZOJ 4715】 树上路径
- 【JZOJ4715】树上路径
- 【JZOJ 4715】树上路径
- 【NOIP提高】树上路径
- 树上最长单色路径
- 循环结构
- 必须要把红黑树讲清楚,看完还不明白请直接找我之3(共四篇)—红黑树的删除
- 带有期限的作业排序java实现
- node.js基础知识 (一)
- 博客
- [JZOJ5055]树上路径
- hdoj-2016-数据的交换输出(解题报告)
- leetcodeOJ 80. Remove Duplicates from Sorted Array II
- 命令行wifi配置 wpa_supplicant 配置与应用
- Window sum
- 搜狗后台测试面试
- JAVA 中的Collection FrameWork
- 翻转二叉树
- 让电脑linux系统,安卓手机开机启动变成文本显示,字符串