JZOJ6341. 【NOIP2019模拟2019.9.4】C

2019-09-10 22:57:49 阅读：249 来源： 互联网

标签：JZOJ6341 2019.9 int maxp NOIP2019 fa ch maxn include

Description

在这里插入图片描述
1<=N,Q,0<=ai<=323232323

Solution

考虑倍增。
设f[i][j]表示从i往上跳2^j^步，从i到fa[i][j]的路径上所有点不包括fa[i][j]，a[x]|dis(x,i)的和。
考虑合并，从f[i][j-1]转移过来。那么问题在于从fa[i][j-1]应该是从2^j-1^开始.
但是f[fa[i][j-1]][j-1]是从0开始往上的。
我们可以注意到f[fa[i][j-1]][j-1]中并没有或过2^j-1^，所以实际上是可以合并的,因为或过的数在二进制上没有交。
所以只需要记录在这一段路径内j-1位0的个数就好了（避免算重）。
同理我们可以计算一个从上往下的相反的倍增数组。
考虑u，v的路径，现计算u--lca,再计算(v的dis(u,v)个祖先)--v，减去(v的dis(u,v)个祖先)--lca
这样容斥一下就好了。
$ O(n log n) $

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define maxn 300005
#define maxp 20
#define ll long long 
using namespace std;

int n,q,i,j,k,a[maxn],x,y;
int em,e[maxn*2],nx[maxn*2],ls[maxn];
int fa[maxn][maxp],dep[maxn],cnt[maxn][maxp];
int u,v,w;
ll g0[maxn][maxp],g1[maxn][maxp];

void read(int &x){
    x=0; char ch=getchar();
    for(;ch<'0'||ch>'9';ch=getchar());
    for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
}

void insert(int x,int y){
    em++; e[em]=y; nx[em]=ls[x]; ls[x]=em;
    em++; e[em]=x; nx[em]=ls[y]; ls[y]=em;
}

void DFS(int x,int p){
    fa[x][0]=p,g0[x][0]=a[x],g1[x][0]=a[fa[x][0]],dep[x]=dep[p]+1;
    for(int i=0;i<maxp;i++) cnt[x][i]=cnt[p][i]+((a[x]>>i)&1^1);
    if (x==3){
        printf("");
    }
    for(int i=1;i<maxp;i++) {
        int y=fa[x][i-1];
        int z=fa[x][i]=fa[y][i-1];
        g0[x][i]=g0[x][i-1]+g0[y][i-1]+(1ll<<i-1)*(cnt[y][i-1]-cnt[z][i-1]);
        g1[x][i]=g1[x][i-1]+g1[y][i-1]+(1ll<<i-1)*(cnt[fa[x][0]][i-1]-cnt[fa[y][0]][i-1]);
    }
    for(int i=ls[x];i;i=nx[i]) if (e[i]!=p)
        DFS(e[i],x);
}

int getlca(int x,int y){
    if (dep[x]<dep[y]) swap(x,y);
    for(int i=maxp-1;i>=0;i--) if (dep[fa[x][i]]>=dep[y])
        x=fa[x][i];
    if (x==y) return x;
    for(int i=maxp-1;i>=0;i--) if (fa[x][i]!=fa[y][i])
        x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}

ll GET0(int x,int k){
    int y=x;
    for(i=maxp-1;i>=0;i--) if (k&(1<<i)) 
        y=fa[y][i];
    ll sum=0;
    for(i=maxp-1;i>=0;i--) if (k&(1<<i)){
        sum+=g0[x][i]+(1ll<<i)*(cnt[fa[x][i]][i]-cnt[y][i]);
        x=fa[x][i];
    }
    return sum;
}

int t,d[maxn];
ll GET1(int x,int k){
    int y=x;
    for(i=0;i<=maxp-1;i++) if (k&(1<<i))
        d[++t]=y,y=fa[y][i];
    
    ll sum=0;
    for(i=maxp-1;i>=0;i--) if (k&(1<<i)){
        sum+=g1[d[t]][i]+(1ll<<i)*(cnt[fa[x][0]][i]-cnt[fa[d[t]][0]][i]);
        t--;
    }
    return sum+(a[x]|k);
}

int main(){
    read(n),read(q);
    for(i=1;i<=n;i++) read(a[i]);
    for(i=1;i<n;i++) read(x),read(y),insert(x,y);
    DFS(1,0);
    while (q--){
        read(u),read(v),w=getlca(u,v);
        printf("%lld\n",GET0(u,dep[u]-dep[w])
                    +GET1(v,dep[u]+dep[v]-2*dep[w])-GET1(w,dep[u]-dep[w])
                    +1ll*(a[w]|(dep[u]-dep[w])));
    }
}

标签：JZOJ6341,2019.9,int,maxp,NOIP2019,fa,ch,maxn,include
来源： https://www.cnblogs.com/DeepThinking/p/11503582.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

JZOJ6341. 【NOIP2019模拟2019.9.4】C

Description

Solution