BZOJ 3365: [Usaco2004 Feb]Distance Statistics 路程统计 点分治

3365: [Usaco2004 Feb]Distance Statistics 路程统计

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 264  Solved: 153

[Submit][Status][Discuss]

Description

    在得知了自己农场的完整地图后（地图形式如前三题所述），约翰又有了新的问题．他提供
一个整数K(1≤K≤109)，希望你输出有多少对农场之间的距离是不超过K的．

Input

    第1到I+M行：与前三题相同；
    第M+2行：一个整数K.

Output

    农场之间的距离不超过K的对数．

Sample Input

7 6

1 6 13 E

6 3 9 E

3 5 7 S

4 1 3 N

2 4 20 W

4 7 2 S

10

Sample Output

5

有五对道路之间的距离小于10

1-4,距离为3

4-7,距离为2

1-7,距离为5

3-5,距离为7

3-6,距离为9

裸点分治

这个题有鬼畜的字符。。。

还 诚惶诚恐地加了输入字符串。。。

我有读入优化啊。。。

笑哭

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
inline int read()
{
int x=0,f=1;char 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 x*f;
}
inline void print(int x)
{
if(x<0)putchar('-'),x=-x;
if(x>=10)print(x/10);putchar('0'+x%10);
}
const int N=40010,inf=0X3f3f3f3f;
int ecnt,last
,k,n,root,sum,size
,f
,t
,dis
,ans;
bool vis
;
struct EDGE{int to,nt,val;}e[N<<1];
inline void add(int u,int v,int val)
{e[++ecnt]=(EDGE){v,last[u],val};last[u]=ecnt;}
void getroot(int u,int fa)
{
f[u]=0;size[u]=1;
for(int i=last[u];i;i=e[i].nt)
{
if(e[i].to==fa||vis[e[i].to])continue;
getroot(e[i].to,u);
size[u]+=size[e[i].to];
f[u]=max(f[u],size[e[i].to]);
}
f[u]=max(f[u],sum-size[u]);
if(f[u]<f[root])root=u;
}
void seek(int u,int fa)
{
t[++t[0]]=dis[u];
for(int i=last[u];i;i=e[i].nt)
{
if(vis[e[i].to]||e[i].to==fa)continue;
dis[e[i].to]=dis[u]+e[i].val;
seek(e[i].to,u);
}
}
inline void cal(int u,int val,int opt)
{
t[0]=0;dis[u]=val;seek(u,0);
sort(t+1,t+1+t[0]);
int l=1,r=t[0];
while(l<r)
{
if(t[l]+t[r]<=k)ans+=opt*(r-l),l++;
else r--;
}
}
void solve(int u)
{
cal(u,0,1);vis[u]=1;
for(int i=last[u];i;i=e[i].nt)
{
if(vis[e[i].to])continue;
cal(e[i].to,e[i].val,-1);root=0;sum=size[e[i].to];
getroot(e[i].to,0);solve(root);
}
}
int main()
{
n=read();int m=read();
while(m--){int u=read(),v=read(),val=read();add(u,v,val);add(v,u,val);}
k=read();f[0]=inf;sum=n;getroot(1,0);solve(root);print(ans);
return 0;
}
/*
7 6
1 6 13 E
6 3 9 E
3 5 7 S
4 1 3 N
2 4 20 W
4 7 2 S
10

5
*/