倍增(RMQ与LCA)模板

来源:互联网 发布:大数据概念龙头股 编辑:程序博客网 时间:2024/05/16 11:49

RMQ

#include<cstdio>#include<cstring>#include<algorithm>#include<cmath>using namespace std;int n,Q,a[100005],Max[100005][20],Min[100005][20];long long sum[100005];void RMQ(){    int t=log(n)/log(2);    for(int i=1;i<=n;i++)    {        Max[i][0]=Min[i][0]=a[i];    }    for(int i=1;i<=t;i++)    {        for(int j=1;j<=n-(1<<i)+1;j++)        {            Max[j][i]=max(Max[j][i-1],Max[j+(1<<(i-1))][i-1]);            Min[j][i]=min(Min[j][i-1],Min[j+(1<<(i-1))][i-1]);        }    }}void solve(){    scanf("%d %d",&n,&Q);    sum[0]=0;    for(int i=1;i<=n;i++)    {        scanf("%d",&a[i]);        sum[i]=sum[i-1]+a[i];    }    RMQ();    for(int i=1;i<=Q;i++)    {        int l,r;        scanf("%d %d",&l,&r);        int t=log(r-l+1)/log(2);        printf("%I64d %d %d\n",sum[r]-sum[l-1],max(Max[l][t],Max[r-(1<<t)+1][t]),min(Min[l][t],Min[r-(1<<t)+1][t]));    }}int main(){    solve();    return 0;}

LCA

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;const int N=50005;int n,Q,head[N],Next[N*2],To[N*2],Cost[N*2],cnt,dep[N],father[N][20],sum[N][20],Max[N][20];void init(){    cnt=-1;    memset(head,-1,sizeof(head));}void addedge(int u,int v,int w){    cnt++;    Next[cnt]=head[u];    To[cnt]=v;    Cost[cnt]=w;    head[u]=cnt;}void dfs(int u){    for(int i=head[u];~i;i=Next[i])    {        int v=To[i];        if(v!=father[u][0])        {            father[v][0]=u;            Max[v][0]=Cost[i];            dep[v]=dep[u]+1;            dfs(v);        }    }}void getfather(){    for(int j=1;j<20;j++) for(int i=1;i<=n;i++)    {        if(father[i][j-1]!=0)        {            father[i][j]=father[father[i][j-1]][j-1];            sum[i][j]=sum[i][j-1]+sum[father[i][j-1]][j-1];            Max[i][j]=max(Max[i][j-1],Max[father[i][j-1]][j-1]);        }        else        {            father[i][j]=0;        }    }}void query(int x,int y,int &A,int &B,int &C){    if(dep[x]<dep[y]) swap(x,y);    for(int i=19;i>=0;i--)    {        if(dep[x]-(1<<i)>=dep[y])        {            B+=sum[x][i];            C=max(C,Max[x][i]);            x=father[x][i];        }    }    if(x==y)    {        A=x;        B+=sum[x][0];        return;    }    for(int i=19;i>=0;i--)    {        if(father[x][i]!=father[y][i])        {            B+=sum[x][i]+sum[y][i];            C=max(C,max(Max[x][i],Max[y][i]));            x=father[x][i];            y=father[y][i];        }    }    A=father[x][0];    B+=sum[x][1]+sum[y][0];    C=max(C,max(Max[x][0],Max[y][0]));}void solve(){    init();    scanf("%d %d",&n,&Q);    for(int i=1;i<n;i++)    {        int u,v,w;        scanf("%d %d %d",&u,&v,&w);        addedge(u,v,w);        addedge(v,u,w);    }    father[1][0]=0;    Max[1][0]=0;    dep[1]=1;    dfs(1);    for(int i=1;i<=n;i++)    {        scanf("%d",&sum[i][0]);    }    getfather();    for(int i=1;i<=Q;i++)    {        int x,y,A=0,B=0,C=0;        scanf("%d %d",&x,&y);        query(x,y,A,B,C);        printf("%d %d %d\n",A,B,C);    }}int main(){    solve();    return 0;}
原创粉丝点击