wannafly挑戰賽4樹的距離 離線處理，dfs序
阿新 • • 發佈：2017-11-28
build void esc ring 空間限制 tps alt 次數 urn 
時間限制：C/C++ 2秒，其他語言4秒
空間限制：C/C++ 262144K，其他語言524288K
64bit IO Format: %lld
題目描述
wyf非常喜歡樹。一棵有根數樹上有N個節點，1號點是他的根，每條邊都有一個距離，而wyf是個愛問奇怪問題的熊孩子，他想知道對於某個點x,以x為根的子樹上，所有與x距離大於等於k的點與x的距離之和。輸入描述:
第一行一個正整數N
接下來N-1描述這棵樹，每行兩個數第i行兩個數p和D表示樹上有一條p到i+1長度為D的邊。(p<=i)
下面一行一個正整數Q表示wyf的詢問次數。
接下來Q行每行兩個正整數x和k。 (1<=N,Q<=2x105 ,1<=D,K<=106)
輸出描述:
對於每次詢問x,k輸出以x為根的子樹上，所有與x距離大於等於k的點與x的距離之和。(若不存在這樣的點，則輸出應為0)示例1
輸入
3 1 2 1 3 2 1 3 1 2
輸出
3 5
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=2e5+88;
long long ans[N];
int tot,head[N],pos[N],kt,lt[N];
struct cta{
int next,to,w;
}e[N<<1];
struct node{
int num,l,r;
long long sum;
}tr[N<<2];
struct qqq{
int id,rto;
long long k;
bool operator <(const qqq &A)const{
return k>A.k;
}
}qt[N];
struct ttt{
long long dis;
int id;
bool operator < (const ttt &B)const{
return dis>B.dis;
}
}dt[N];
void add(int u,int v,int w) {
e[tot].to=v;e[tot].next=head[u];e[tot].w=w;head[u]=tot++;
}
void build(int l,int r,int rt){
tr[rt].l=l,tr[rt].r=r;
tr[rt].sum=tr[rt].num=0;
if(l==r) return;
int mid=(l+r)>>1;
build(l,mid,rt<<1);
build(mid+1,r,rt<<1|1);
}
void dfs(int u,long long now,int f){
pos[u]=++kt;
for(int i=head[u];~i;i=e[i].next) {
if(e[i].to==f) continue;
else dfs(e[i].to,dt[e[i].to].dis=e[i].w+now,u);
}
lt[u]=kt;
}
void modify(int pos,int rt,int cc){
int l=tr[rt].l,r=tr[rt].r;
int mid=(l+r)>>1;
if(l==r) {
tr[rt].num=1;
tr[rt].sum+=dt[cc].dis;
return;}
if(pos<=mid) modify(pos,rt<<1,cc);
else modify(pos,rt<<1|1,cc);
tr[rt].num=tr[rt<<1].num+tr[rt<<1|1].num;
tr[rt].sum=tr[rt<<1].sum+tr[rt<<1|1].sum;
}
long long query(int L,int R,int rt,long long &cty){
long long now=0;
if(L>R) return 0;
int l=tr[rt].l,r=tr[rt].r;
if(L<=l&&R>=r) {cty+=tr[rt].num; return tr[rt].sum;}
int mid=(l+r)>>1;
if(mid>=L) now+=query(L,R,rt<<1,cty);
if(mid<R) now+=query(L,R,rt<<1|1,cty);
return now;
}
int tran[N];
int main(){
int n,x,y,m;
memset(head,-1,sizeof(head));
tot=kt=0;
scanf("%d",&n);
build(1,n,1);
for(int i=2;i<=n;++i) {
scanf("%d%d",&x,&y);
add(x,i,y);
add(i,x,y);
}
dfs(1,0,0);
for(int i=1;i<=n;++i) dt[i].id=i;
scanf("%d",&m);
for(int i=1;i<=m;++i) {
scanf("%d%lld",&qt[i].rto,&qt[i].k);
qt[i].id=i;
qt[i].k+=dt[qt[i].rto].dis;
}
sort(qt+1,qt+1+m);
sort(dt+1,dt+n+1);
int r=1;
for(int i=1;i<=n;++i) tran[dt[i].id]=i;
for(int i=1;i<=m;++i) {
long long now=0;
for(;r<=n;++r) if(!(dt[r].dis>=qt[i].k)) break;else modify(pos[dt[r].id],1,r);
ans[qt[i].id]=query(pos[qt[i].rto]+1,lt[qt[i].rto],1,now);
ans[qt[i].id]-=now*dt[tran[qt[i].rto]].dis;
}
for(int i=1;i<=m;++i) printf("%lld\n",ans[i]);
}
wannafly挑戰賽4樹的距離 離線處理，dfs序