update 2019.9.29
分解成为\(u->lca\)和\(lca->v\)
得到两个式子,需要统计子树内部的信息。
用差分维护下,差分维护类似于这个前置题。
#include <bits/stdc++.h>
using namespace std;
const int _=5e5+7;
int read() {
int x=0,f=1;char s=getchar();
for(;s>'9'||s<'0';s=getchar()) if(s=='-') f=-1;
for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';
return x*f;
}
int n,m,w[_],S[_],T[_],lca[_],tong[_*3],ans[_];
struct node {int v,nxt,q;}e[_<<1];
struct edge {int x,y;edge(int a=0,int b=0) {x=a,y=b;}};
vector<edge> ad[_*4];
int head[_],tot;
void add(int u,int v) {
e[++tot].v=v;
e[tot].nxt=head[u];
head[u]=tot;
}
int son[_],dep[_],fa[_],siz[_],top[_];
void dfs1(int u,int f) {
fa[u]=f;
dep[u]=dep[f]+1;
siz[u]=1;
for(int i=head[u];i;i=e[i].nxt) {
int v=e[i].v;
if(v==f) continue;
dfs1(v,u);
siz[u]+=siz[v];
if(siz[son[u]]<siz[v]) son[u]=v;
}
}
void dfs2(int u,int topf) {
top[u]=topf;
if(!son[u]) return;
dfs2(son[u],topf);
for(int i=head[u];i;i=e[i].nxt) {
int v=e[i].v;
if(!top[v]) dfs2(v,v);
}
}
int LCA(int x,int y) {
while(top[x]!=top[y]) {
if(dep[top[x]]<dep[top[y]]) swap(x,y);
x=fa[top[x]];
} return dep[x]<dep[y]?x:y;
}
void solve1(int u,int fa) {
int A=tong[dep[u]+w[u]];
for(int i=0;i<(int)ad[u].size();++i)
tong[ad[u][i].x]+=ad[u][i].y;
for(int i=head[u];i;i=e[i].nxt)
if(e[i].v!=fa) solve1(e[i].v,u);
int B=tong[dep[u]+w[u]];
ans[u]+=B-A;
}
void solve2(int u,int fa) {
int A=tong[w[u]-dep[u]+2*n];
for(int i=0;i<(int)ad[u].size();++i)
tong[ad[u][i].x]+=ad[u][i].y;
for(int i=head[u];i;i=e[i].nxt)
if(e[i].v!=fa) solve2(e[i].v,u);
int B=tong[w[u]-dep[u]+2*n];
ans[u]+=B-A;
}
int main() {
n=read(),m=read();
for(int i=1;i<n;++i) {
int u=read(),v=read();
add(u,v),add(v,u);
}
for(int i=1;i<=n;++i) w[i]=read();
dfs1(1,0),dfs2(1,1);
for(int i=1;i<=m;++i) S[i]=read(),T[i]=read(),lca[i]=LCA(S[i],T[i]);
for(int i=1;i<=m;++i) {
ad[S[i]].push_back(edge(dep[S[i]],1));
ad[fa[lca[i]]].push_back(edge(dep[S[i]],-1));
}
solve1(1,0);
for(int i=1;i<=n*3;++i) ad[i].clear();
for(int i=1;i<=m;++i) {
ad[lca[i]].push_back(edge(dep[S[i]]-2*dep[lca[i]]+2*n,-1));
ad[T[i]].push_back(edge(dep[S[i]]-2*dep[lca[i]]+2*n,1));
}
solve2(1,0);
for(int i=1;i<=n;++i) printf("%d ",ans[i]);
return 0;
}
lca真心不太会，这里只介绍60分做法，100的太难辣简单了就不介绍了
n<=1000
zz回溯爆搜
S[i]全部相等
这dfs序都不用lca的，2333，差分，然后输出判断一下是否是0(1到i的时间是固定的)
退化成一条链子
一个点i的ans就是i-time[i]和i+tim[i]的起点个数(当然要合法啦)
乱搞就好了，这里写的nlogn的
#include <iostream>
#include <queue>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=300000;
const int maxn_25=1000;
struct edge {
int v,nxt;
}e[maxn<<1];
struct node {
int x,id;
bool operator < (const node b) const {
return x>b.x;
}
};
struct node1 {
int x,id;
bool operator < (const node1 b) const {
return x<b.x;
}
};
int head[maxn<<1],tot;
int tim[maxn],n,m,S[maxn],T[maxn],deep[maxn<<1];
int pd_25,ans_25[maxn_25]; //baoli 25 zhuanyong
int tot_a,a_c[maxn<<1],find_a[maxn],ans_c[maxn];//all s == 1 zhuanyong
int vis[maxn<<1],cf[maxn<<1],dsr[maxn<<1];
int ans[maxn];
void add_edge(int u,int v) {
e[++tot].v=v;
e[tot].nxt=head[u];
head[u]=tot;
}
inline int read() {
int x=0,f=1;char s=getchar();
while(s<'0'||s>'9')
{if(s=='-')f=-1;s=getchar();}
while(s>='0'&&s<='9')
{x=(x<<3)+(x<<1)+s-'0',s=getchar();}
return x*f;
}
void dfs_25(int u,int f,int end,int cnt) {
if(u==end) {
if(cnt==tim[u]) ans_25[u]++;
pd_25=1;
return;
}
if(pd_25) return;
if(cnt==tim[u]) ans_25[u]++;
for(int i=head[u];i;i=e[i].nxt) {
int v=e[i].v;
if(v==f) continue;
dfs_25(v,u,end,cnt+1);
if(pd_25) return;
}
if(cnt==tim[u]) ans_25[u]--;
}
void dfs_deep(int u,int f,int cnt) {
deep[u]=cnt;
for(int i=head[u];i;i=e[i].nxt) {
int v=e[i].v;
if(f==v) continue;
dfs_deep(v,u,cnt+1);
}
}
void dfs_xu(int u,int f) {
a_c[++tot_a]=0;
find_a[u]=tot_a;
for(int i=head[u];i;i=e[i].nxt) {
int v=e[i].v;
if(v==f) continue;
dfs_xu(v,u);
}
a_c[++tot_a]=0;
vis[tot_a]=u;
dsr[u]=tot_a;
}
int main() {
n=read(),m=read();
for(int i=1;i<n;++i) {
int x=read(),y=read();
add_edge(x,y);
add_edge(y,x);
}
for(int i=1;i<=n;++i)
tim[i]=read();
for(int i=1;i<=m;++i)
S[i]=read(),T[i]=read();
//baoli_25
if(n<=1000) {
for(int i=1;i<=m;++i) {
pd_25=0;
dfs_25(S[i],0,T[i],0);
}
for(int j=1;j<=n;++j)
printf("%d ",ans_25[j]);
return 0;
}
int wakaka=1;
for(int i=1;i<=m;++i)
if(S[i]!=1) wakaka=0;
if(wakaka)
{
//all S[i]==1 fen==20
dfs_deep(1,0,0);//1 -> i deep
dfs_xu(1,0);
//chafen
for(int i=1;i<=m;++i) {
//1->i
cf[find_a[S[i]]]++;
cf[find_a[T[i]]+1]--;
}
//qiouhe
for(int i=1;i<=tot_a;++i) {
a_c[i]=a_c[i-1]+cf[i];
}
for(int i=1;i<=n;++i) {
if(deep[i]==tim[i])
printf("%d ",a_c[find_a[i]]-a_c[dsr[i]]);
else
printf("0 ");
}
return 0;
}
priority_queue<node> q;
for(int i=1;i<=m;++i)
{
if(S[i] > T[i]) continue;
dsr[S[i]]++;
node x;
x.x=T[i]+1;
x.id=S[i];
q.push(x);
}
for(int i=1;i<=n;++i)
{
while(!q.empty()&&q.top().x==i)
dsr[q.top().id]--,q.pop();
if(i-tim[i]>=1)
ans[i]+=dsr[i-tim[i]];
}
priority_queue< node1 > qq;
memset(dsr,0,sizeof(dsr));
for(int i=1;i<=m;++i)
{
if(S[i] <= T[i]) continue;
dsr[S[i]]++;
node1 x;
x.x=T[i]-1;
x.id=S[i];
qq.push(x);
}
for(int i=n;i>=1;--i)
{
while(!qq.empty()&&qq.top().x==i)
dsr[qq.top().id]--,qq.pop();
// puts("debug");
// for(int j=1;j<=n;++j)
// cout<<dsr[j]<<" ";puts("");
if(i+tim[i]<=n)
ans[i]+=dsr[i+tim[i]];
}
for(int i=1;i<=n;++i)
printf("%d ",ans[i]);
return 0;
}