题目大意:同POJ1741
链接:http://blog.csdn.net/popoqqq/article/details/39959803
和POJ1741一样的题,土豪题,POJ有道一样的题,做完1741可以水水
我会告诉你我直接改了下上题的代码么0.0
数据范围4W 距离最大1000 不会爆INT 放心写吧
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define M 40400 using namespace std; struct abcd{ int to,f,next; bool ban; }table[M<<1]; int head[M],tot=1; int n,m,k,ans; int siz[M],fa[M],dist[M]; int stack[M],top; void Add(int x,int y,int z) { table[++tot].to=y; table[tot].f=z; table[tot].next=head[x]; head[x]=tot; table[tot].ban=0; } void Get_Centre_Of_Gravity(int x,int size,int &cg) { int i; bool flag=1; siz[x]=1; for(i=head[x];i;i=table[i].next) if(!table[i].ban) if(table[i].to!=fa[x]) { fa[table[i].to]=x; Get_Centre_Of_Gravity(table[i].to,size,cg); if(siz[table[i].to]>size>>1) flag=0; siz[x]+=siz[table[i].to]; } if(size-siz[x]>size>>1) flag=0; if(flag) cg=x; } void DFS1(int x,int dis) { int i; dist[x]=dis; for(i=head[x];i;i=table[i].next) { if(table[i].ban) continue; if(table[i].to==fa[x]) continue; fa[table[i].to]=x; DFS1(table[i].to,dis+table[i].f); } } void DFS2(int x) { int i; stack[++top]=dist[x]; for(i=head[x];i;i=table[i].next) { if(table[i].ban) continue; if(table[i].to==fa[x]) continue; DFS2(table[i].to); } } int Calculate(int root) { int i,j,re=0; top=0; DFS2(root); sort(stack+1,stack+top+1); for(i=1,j=top;i<=top;i++) { for(;j&&stack[j]+stack[i]>k;j--); re+=j; } return re; } void Tree_Divide_And_Conquer(int root,int size) { int i,cg; fa[root]=0; Get_Centre_Of_Gravity(root,size,cg); siz[fa[cg]]=size-siz[cg]; fa[cg]=0; DFS1(cg,0); ans+=Calculate(cg); for(i=head[cg];i;i=table[i].next) if(!table[i].ban) { table[i].ban=table[i^1].ban=1; ans-=Calculate(table[i].to); Tree_Divide_And_Conquer(table[i].to,siz[table[i].to]); } } int main() { int i,x,y,z; char p[10]; cin>>n>>m; for(i=1;i<n;i++) scanf("%d%d%d%s",&x,&y,&z,p),Add(x,y,z),Add(y,x,z); cin>>k; Tree_Divide_And_Conquer(1,n); cout<<(ans-n>>1)<<endl; }
时间: 2024-10-28 19:18:44