[SDOI2019]世界地图(kruskal重构树+虚树)

通过子任务1、3十分显然,子任务4实际上就是线段树,和我下午写的[SDOI2015]道路修建一模一样,堪称“我抄我自己”,不会的可以先做一下这个题。

然后考虑正解,参考了zhoushuyu的博客,首先可以对前i列做MST,就是把前i-1列和第i列合并起来,而这时候只需要把第1和第i列的点作为关键点建立虚树,虚树边权为原树路径最大值,然后每次O(n)对虚树合并即可。后缀也同样做一遍即可。查询时,就是把整张图分成两半,同样只需要维护前后缀的左右两列建立虚树即可,复杂度O(n(m+q)logn)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=10086;
struct edge{int u,v,w;};
bool operator<(edge a,edge b){return a.w<b.w;}
int n,m,Q,lim,tot,cnt,d1[N][110],d2[N][110],fa[N],hd[N],v[N],nxt[N],w[N],vis[N];
ll ans;
unsigned int SA,SB,SC;
vector<edge>G;
struct MST{
    int tot;ll sum;
    vector<edge>G;
    MST(){}
    MST(int*c)
    {
        tot=n,sum=0;
        for(int i=1;i<n;i++)G.push_back((edge){i,i+1,c[i]});
    }
}pre[N],suf[N];
int getweight()
{
    SA^=SA<<16,SA^=SA>>5,SA^=SA<<1;
    unsigned int t=SA;
    SA=SB,SB=SC,SC^=t^SA;
    return SC%lim+1;
}
int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
void adde(edge x)
{
    v[++cnt]=x.v,nxt[cnt]=hd[x.u],w[cnt]=x.w,hd[x.u]=cnt;
    v[++cnt]=x.u,nxt[cnt]=hd[x.v],w[cnt]=x.w,hd[x.v]=cnt;
    ans+=x.w;
}
bool dfs1(int u,int f)
{
    int ret=0;
    for(int i=hd[u];i;i=nxt[i])if(v[i]!=f)ret+=dfs1(v[i],u);
    vis[u]|=(ret>=2),ret+=vis[u];
    return ret;
}
void dfs2(int u,int f,int lst,int val)
{
    if(vis[u])
    {
        if(lst)G.push_back((edge){vis[u],lst,val});
        lst=vis[u],ans-=val,val=0;
    }
    for(int i=hd[u];i;i=nxt[i])if(v[i]!=f)dfs2(v[i],u,lst,max(val,w[i]));
}
MST merge(MST a,MST b,int*c)
{
    G.clear(),tot=a.tot+b.tot;
    for(int i=0;i<a.G.size();i++)G.push_back(a.G[i]);
    for(int i=0;i<b.G.size();i++)G.push_back((edge){b.G[i].u+a.tot,b.G[i].v+a.tot,b.G[i].w});
    for(int i=1;i<=n;i++)G.push_back((edge){a.tot-n+i,a.tot+i,c[i]});
    sort(G.begin(),G.end());
    for(int i=1;i<=tot;i++)fa[i]=i,vis[i]=(i<=n||i>tot-n),hd[i]=0;
    ans=cnt=0;
    for(int i=0;i<G.size();i++)
    {
        int u=find(G[i].u),v=find(G[i].v);
        if(u!=v)adde(G[i]),fa[u]=v;
    }
    dfs1(1,0),cnt=0;
    for(int i=1;i<=tot;i++)if(vis[i])vis[i]=++cnt;
    G.clear(),dfs2(1,0,0,0);
    MST ret;ret.tot=cnt,ret.sum=a.sum+b.sum+ans,ret.G=G;
    return ret;
}
ll query(MST a)
{
    ll ret=a.sum;
    for(int i=0;i<a.G.size();i++)ret+=a.G[i].w;
    return ret;
}
int main()
{
    scanf("%d%d%u%u%u%d",&n,&m,&SA,&SB,&SC,&lim);
    for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)d1[j][i]=getweight();
    for(int i=1;i<n;i++)for(int j=1;j<=m;j++)d2[j][i]=getweight();
    pre[1]=MST(d2[1]),suf[m]=MST(d2[m]);
    for(int i=2;i<m;i++)pre[i]=merge(pre[i-1],MST(d2[i]),d1[i-1]);
    for(int i=m-1;i>1;i--)suf[i]=merge(MST(d2[i]),suf[i+1],d1[i]);
    scanf("%d",&Q);
    while(Q--)
    {
        int l,r;scanf("%d%d",&l,&r);
        printf("%lld\n",query(merge(suf[r+1],pre[l-1],d1[m])));
    }
}
View Code

 

posted @ 2019-05-07 22:43  hfctf0210  阅读(650)  评论(0编辑  收藏  举报