Hanks_o的博客

有权限号的bzoj蒟蒻用户

bzoj4448: [Scoi2015]情报传递(主席树+Lca)

题目传送门
严重吐槽这题数据!!
花了一下午一直在调一个AC代码。
提交莫名WA。因为我建的是单向边。
题目说明规定了父亲。那不就是单向边吗。。
然后本机栈不够大RE了半天。
然后肉老师告诉我一个本地扩栈的方法然后我过了。。
然后又WA了。
然后%题解。发现建的都是双向边。
这题不是规定了父亲吗???
然后建了双向才过了。。。
数据跟题意不符啊。

解法:
当前第i天有个询问。问到了某个节点,这个节点是从第j天开始收集情报的。
那么当i-j>c才能构成威胁。
转化一下i-c>j。
那么每个节点的权值就为这个点什么时候开始收集情报。
那么每次求比i-c小的节点个数就行了。
每个点的主席树维护根到这个点的权值。
然后lca搞一下即可。

代码实现:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
struct node {int c,lc,rc;}t[11000000];int cnt,rt[2100000];
void build(int &u,int l,int r,int p) {
    if(u==0)u=++cnt;t[u].c++;
    if(l==r)return ;int mid=(l+r)/2;
    if(p<=mid)build(t[u].lc,l,mid,p);
    else build(t[u].rc,mid+1,r,p);
}
void Merge(int &u1,int u2) {
    if(u1==0) {u1=u2;return ;}if(u2==0)return ;
    t[u1].c+=t[u2].c;
    Merge(t[u1].lc,t[u2].lc);Merge(t[u1].rc,t[u2].rc);
}
struct trnode {int x,y,next;}a[2100000];int len,last[2100000];
void ins(int x,int y) {len++;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;}
int z,dep[2100000],mx[21][210000];
void dfs(int x) {
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(y!=mx[0][x]) {mx[0][y]=x;dep[y]=dep[x]+1;dfs(y);}
    }
}
int bin[21],n;
void work() {
    bin[0]=1;for(int i=1;i<=20;i++)bin[i]=bin[i-1]*2;
    for(int j=1;j<=20;j++)for(int i=1;i<=n;i++)if(dep[i]>=bin[j])
        mx[j][i]=mx[j-1][mx[j-1][i]];
}
int lca(int x,int y) {
    if(dep[x]>dep[y])swap(x,y);
    for(int i=20;i>=0;i--)if(dep[y]-dep[x]>=bin[i])y=mx[i][y];
    if(x==y)return x;
    for(int i=20;i>=0;i--)if(dep[x]>=bin[i]&&mx[i][x]!=mx[i][y])
        x=mx[i][x],y=mx[i][y];
    return mx[0][x];
}
int find(int u1,int u2,int u3,int u4,int l,int r,int p) {
    if(l==r)return t[u1].c+t[u2].c-t[u3].c-t[u4].c;
    int mid=(l+r)/2;
    if(p<=mid)return find(t[u1].lc,t[u2].lc,t[u3].lc,t[u4].lc,l,mid,p);
    else return find(t[u1].rc,t[u2].rc,t[u3].rc,t[u4].rc,mid+1,r,p)+
        t[t[u1].lc].c+t[t[u2].lc].c-t[t[u3].lc].c-t[t[u4].lc].c;
}
struct edge {int t,x,y,c;}s[210000];
void merge(int x) {
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(y!=mx[0][x]){Merge(rt[y],rt[x]);merge(y);}
    }
}
int main() {
    //freopen("4448.in","r",stdin);freopen("4448.out","w",stdout);
    scanf("%d",&n);int root;len=0;memset(last,0,sizeof(last));cnt=0;
    for(int i=1;i<=n;i++) {int x;scanf("%d",&x);if(x==0)root=i;else ins(x,i);}
    int Q;dep[root]=0;mx[0][root]=0;z=0;dfs(root);work();scanf("%d",&Q);
    for(int i=1;i<=Q;i++) {
        scanf("%d%d",&s[i].t,&s[i].x);
        if(s[i].t==2)build(rt[s[i].x],1,Q,i);
        else scanf("%d%d",&s[i].y,&s[i].c);
    }merge(root);
    for(int i=1;i<=Q;i++)if(s[i].t==1) {
        int Lca=lca(s[i].x,s[i].y);
        printf("%d %d\n",dep[s[i].x]+dep[s[i].y]-dep[Lca]*2+1,
            find(rt[s[i].x],rt[s[i].y],rt[Lca],rt[mx[0][Lca]],1,Q,i-s[i].c-1));
    }
    return 0;
}
阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Hanks_o/article/details/79979199
个人分类: BZOJ LCA 主席树
上一篇bzoj2733: [HNOI2012]永无乡(并查集+主席树)
下一篇bzoj3123: [Sdoi2013]森林(主席树+Lca+启发式合并)
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭