hdu5111
链接
hdu
思路
先考虑序列上如何解决。
1 3 2 5 4
1 2 4 5 3这个序列变成
1 2 3 4 5
1 3 5 5 2是对答案没有影响的(显然)。
然后查询操作\(l,r,L,R\)就是，
一段连续的区间\([L,R]\)内包含几个值在\([l,r]\)的数字个数.
主席树就可以做了。
\(query(rt[L-1],rt[R],[l,r]的和)\)
可以用树链剖分把树上问题转化成链上。
左边一棵树树链剖分，每一条链子都是一段连续的。
右边一棵树根据父子关系建立主席树。
然后向上跳统计贡献。
吐槽
本来我写完之后可以两遍编译，一遍样例，一遍AC的。
鬼知道我清空出了什么毛病。
for(i 0 to limit-1) a[i]=0居然清空不了。
debug了半天，换成了memset才过。
我也是醉了。
代码
#include <iostream>
#include <map>
#include <cstring>
#include <algorithm>
#define ls(x) (t[x].ls)
#define rs(x) (t[x].rs)
using namespace std;
const int _=1e5+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 n1,n2,w1[_],w2[_],rt[_];
map<int,int> dsr;
struct node {int v,nxt;
}e1[_<<1],e2[_<<1];
int head1[_],head2[_],tot1,tot2;
void add1(int u,int v) {e1[++tot1].v=v;e1[tot1].nxt=head1[u];head1[u]=tot1;
}
void add2(int u,int v) {e2[++tot2].v=v;e2[tot2].nxt=head2[u];head2[u]=tot2;
}
namespace ZXS {struct node {int ls,rs,tot;}t[_*30];int cnt;void clear() {memset(t,0,sizeof(t));// for(int i=0;i<=cnt;++i)// ls(i)=rs(i)=t[i].tot=0;cnt=0;}void insert(int l,int r,int k,int x,int &y) {t[y=++cnt]=t[x];t[y].tot++;if(l==r) return;int mid=(l+r)>>1;if(k<=mid) insert(l,mid,k,ls(x),ls(y));else insert(mid+1,r,k,rs(x),rs(y));}int query(int l,int r,int L,int R,int x,int y) {if(L<=l&&r<=R) return t[y].tot-t[x].tot;int mid=(l+r)>>1,ans=0;if(L<=mid) ans+=query(l,mid,L,R,ls(x),ls(y));if(R>mid) ans+=query(mid+1,r,L,R,rs(x),rs(y));return ans;}
}
namespace LCA{int siz[_],son[_],top[_],f[_],dep[_],idx[_],cnt;void clear() {cnt=0;memset(idx,0,sizeof(idx));// memset(siz,0,sizeof(siz));memset(son,0,sizeof(son));// memset(top,0,sizeof(top));// memset(dep,0,sizeof(dep));// memset(f,0,sizeof(f));}void dfs1(int u,int fa) {dep[u]=dep[fa]+1;f[u]=fa;siz[u]=1;for(int i=head2[u];i;i=e2[i].nxt) {int v=e2[i].v;if(v==fa) continue;dfs1(v,u);siz[u]+=siz[v];if(siz[v]>siz[son[u]]) son[u]=v;}}void dfs2(int u,int topf) {idx[u]=++cnt;top[u]=topf;if(!son[u]) return;dfs2(son[u],topf);for(int i=head2[u];i;i=e2[i].nxt) {int v=e2[i].v;if(!idx[v]) dfs2(v,v);}}int query(int x,int y) {while(top[x]!=top[y]) {if(dep[top[x]]<dep[top[y]]) swap(x,y);x=f[top[x]];}if(dep[x]>dep[y]) swap(x,y);return x;}
}
namespace SLPF{int siz[_],son[_],top[_],f[_],dep[_],idx[_],cnt;void clear() {cnt=0;memset(idx,0,sizeof(idx));// memset(siz,0,sizeof(siz));memset(son,0,sizeof(son));// memset(top,0,sizeof(top));// memset(dep,0,sizeof(dep));// memset(f,0,sizeof(f));}void dfs1(int u,int fa) {dep[u]=dep[fa]+1;f[u]=fa;siz[u]=1;for(int i=head1[u];i;i=e1[i].nxt) {int v=e1[i].v;if(v==fa) continue;dfs1(v,u);siz[u]+=siz[v];if(siz[v]>siz[son[u]]) son[u]=v;}}void dfs2(int u,int topf) {idx[u]=++cnt;dsr[w1[u]]=cnt;top[u]=topf;if(!son[u]) return;dfs2(son[u],topf);for(int i=head1[u];i;i=e1[i].nxt) {int v=e1[i].v;if(!idx[v]) dfs2(v,v);}}void QQ(int x,int y,int u2,int v2) {int lca=LCA::query(u2,v2),ans=0;while(top[x]!=top[y]) {if(dep[top[x]]<dep[top[y]]) swap(x,y);ans+=ZXS::query(1,n1,idx[top[x]],idx[x],rt[lca],rt[u2]);// cout<<ZXS::query(1,n1,idx[top[x]],idx[x],rt[lca],rt[u2])<<"\n";ans+=ZXS::query(1,n1,idx[top[x]],idx[x],rt[LCA::f[lca]],rt[v2]);// cout<<ZXS::query(1,n1,idx[top[x]],idx[x],rt[LCA::f[lca]],rt[v2])<<"\n";x=f[top[x]];}if(dep[x]>dep[y]) swap(x,y);ans+=ZXS::query(1,n1,idx[x],idx[y],rt[lca],rt[u2]);// cout<<ZXS::query(1,n1,idx[x],idx[y],rt[lca],rt[u2])<<"\n";ans+=ZXS::query(1,n1,idx[x],idx[y],rt[LCA::f[lca]],rt[v2]);// cout<<ZXS::query(1,n1,idx[x],idx[y],rt[LCA::f[lca]],rt[v2])<<"\n";printf("%d\n",ans);}
}
void dfs(int u,int fa) {if(dsr.count(w2[u])) ZXS::insert(1,n1,dsr[w2[u]],rt[fa],rt[u]);else rt[u]=rt[fa];for(int i=head2[u];i;i=e2[i].nxt) {int v=e2[i].v;if(v==fa) continue;dfs(v,u);}
}
int main() {// freopen("data.in","r",stdin);// freopen("a.out","w",stdout);while(scanf("%d",&n1)!=EOF) {//clearLCA::clear();SLPF::clear();ZXS::clear();dsr.clear();tot1=tot2=0;memset(rt,0,sizeof(rt));memset(head1,0,sizeof(head1));memset(head2,0,sizeof(head2));//readfor(int i=2;i<=n1;++i) {int u=i,v=read();
// cout<<u<<"->"<<v<<"\n";add1(u,v),add1(v,u);}for(int i=1;i<=n1;++i) w1[i]=read();n2=read();for(int i=2;i<=n2;++i) {int u=i,v=read();
// cout<<u<<"->"<<v<<"\n";add2(u,v),add2(v,u);}for(int i=1;i<=n2;++i) w2[i]=read();//initSLPF::dfs1(1,0);SLPF::dfs2(1,1);LCA::dfs1(1,0);LCA::dfs2(1,1);dfs(1,0);//askint Q=read();while(Q --> 0) {int u1=read(),v1=read(),u2=read(),v2=read();SLPF::QQ(u1,v1,u2,v2);}}return 0;
}
)
)
