ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

[TJOI2019] 甲苯先生的线段树

2019-05-08 15:42:18  阅读:268  来源: 互联网

标签:h0 la int 线段 h1 lca TJOI2019 甲苯 LL


臭名昭著的巧合:CF750G

题意:在无限深度的一颗线段树中询问编号和为S的简单路径条数。

题解传送门

这道题相当于在原来基础上多了询问两点间简单路径的编号的的问题。

直觉告诉我们只需要求出两点在线段树上的lca,然后套用上个问题中所推得的式子即可。而线段树上两点的lca的二进制表示正好是两点的二进制表示的lcp,这玩意儿瞎写即可。

参考代码

#include <bits/stdc++.h>
#define LL long long 
using namespace std;
const int N=52;

inline LL calcS(LL a,LL b) {
    LL lca=a,tmp=b,sum=0;
    int la=log2(a),lb=log2(b);
    for(int i=la; i<lb; ++i) tmp>>=1;
    for(int i=lb; i<la; ++i) lca>>=1;
    while(tmp!=lca) tmp>>=1,lca>>=1;
    int LLCA=log2(lca);
    if(a==lca&&b==lca) return lca;
    if(a==lca||b==lca) {
        if(b==lca) a^=b^=a^=b,la^=lb^=la^=lb;
        sum=lca*((1LL<<(lb-LLCA+1))-1);
        for(int i=1; b>lca; ++i,b>>=1) sum+=(b&1)*((1LL<<i)-1);
    } else {
        if((a>>(la-LLCA-1))&1) a^=b^=a^=b,la^=lb^=la^=lb;
        sum=lca*((1LL<<(la-LLCA+1))+(1LL<<(lb-LLCA+1))-3)+(1LL<<(lb-LLCA))-1;
        lca=lca<<1|1;
        for(int i=1; a>lca; ++i,a>>=1) sum+=(a&1)*((1LL<<i)-1);
        for(int i=1; b>lca; ++i,b>>=1) sum+=(b&1)*((1LL<<i)-1);
    }
    return sum;
}
LL P[N]={1},f[N][N*2][2];
inline LL calcP(LL S,int d) {
    LL ans=0;
    int L=min((int)log2(S+1),d);
    for(int h=1; h<=L; ++h) {
        LL x=S/(P[h]-1);
        if(h+(int)log2(x)>d) continue;
        x=S%(P[h]-1);
        for(int i=h; i; --i) if(x>=P[i]-1) x-=P[i]-1;
        ans+=(!x);
    }
    for(int h0=1; h0<L; ++h0) 
    for(int h1=1; S+1-P[h1]>=P[h0+1]+P[h1+1]-3; ++h1) {
        LL x=(S+1-P[h1])/(P[h0+1]+P[h1+1]-3);
        LL r=(S+1-P[h1])%(P[h0+1]+P[h1+1]-3);
        if(max(h0,h1)+1+(int)log2(x)>d) continue;
        if(!r) {ans++; continue;}
        if(h0==1&&h1==1) {ans+=(S==x*5+1); continue;}
        for(int n=1; n<=h0+h1; ++n) {
            LL C=r+n,L=log2(C);
            if(C&1) continue;
            memset(f[0],0,sizeof f[0]);
            f[0][0][0]=1;
            for(int i=1; i<=L; ++i) {
                int d=(C>>i)&1;
                memset(f[i],0,sizeof f[i]);
                for(int j=0; j<=i+i-2&&j<=n; ++j) 
                for(int k=0; k<2; ++k) if(f[i-1][j][k]) 
                    for(int x=0; x<2; ++x) if(!x||i<h0) 
                    for(int y=0; y<2; ++y) if(!y||i<h1) 
                        if(((k+x+y)&1)==d) f[i][j+x+y][(k+x+y)>>1]+=f[i-1][j][k];
            }
            ans+=f[L][n][0];
        }
    }
    return ans;
}

int main() {
    int T;
    scanf("%d",&T);
    for(int i=1; i<N; ++i) P[i]=P[i-1]<<1;
    for(LL c,a,b,d; T--; ) {
        scanf("%lld%lld%lld%lld",&d,&a,&b,&c);
        if(c==1) printf("%lld\n",calcS(a,b));
        else printf("%lld\n",calcP(calcS(a,b),d)-1);
    }
    return 0;
}

标签:h0,la,int,线段,h1,lca,TJOI2019,甲苯,LL
来源: https://www.cnblogs.com/nosta/p/10832146.html

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

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有