莫队

最新推荐文章于 2024-03-31 14:08:11 发布

csu_xiji

最新推荐文章于 2024-03-31 14:08:11 发布

阅读量175

点赞数

分类专栏：莫队

本文链接：https://blog.csdn.net/xiji333/article/details/105179574

版权

莫队专栏收录该内容

8 篇文章 0 订阅

订阅专栏

关键点：

在这里插入图片描述
$1 .$ 知道了 $[l, r]$ 的结果，可以在 $O (1)$ 时间内得到 $[l - 1, r] 、 [l + 1, r] 、 [l, r - 1] 、 [l, r + 1]$ 的结果。
$2 .$ 一般是用来离线计算答案的。
$3 .$ 分块 $+$ 排序保证了时间复杂度。

树上莫队：

因为普通的问题都是直接对区间进行操作的，而树上问题往往是对链或者子树进行操作的，这时候就需要欧拉序和 $d f s$ 序了。
$t i p s ：$ 建议先了解欧拉序和 $d f s$ 序之后再往下看。

对链进行询问

如果我们有办法把链转化成区间，那么问题不就解决了吗？树上莫队就是基于这样的思想，通过欧拉序把链转换成对应的区间，还需要 $L C A$ 的知识，详情可以在上面推荐的博客中学习，这里简单说一下。
在这里插入图片描述
欧拉序列其实非常简单，就是在 $d f s$ 进入时和退出时分别记录该节点的编号，比如上面这棵树对应的欧拉序列就是：

1	2	3	4	5	6	7	8	9	10	11	12	13	14	15	16	17	18
1	2	4	7	7	4	5	5	2	3	6	8	9	9	8	6	3	1

第一行是欧拉序的编号，第二行是对应的节点编号。我们用 $i n [i]$ 表示第一次进入节点 $i$ 时的编号， $o u t [i]$ 表示从节点 $i$ 退出时的编号。那么我们可以发现区间 $[i n [i], o u t [i]]$ 对应了 $i$ 及 $i$ 的子树，且每个节点都出现了两次。然后我们来看一下怎么把链进行转化。先举一个简单的例子，比如 $2 - 7$ 的链，我们知道 $L C A (2, 7) = 2$ ，所以对应的区间就是 $[i n [2], i n [7]]$ ，也就是 $[2, 4]$ ，可以看到对应的编号为 $2 、 4 、 7$ ；再看一个复杂的例子，比如 $4 - 3$ 的链，因为 $L C A (4, 3) = 1$ ，不等于 $4$ 也不等于 $3$ ，所以对应的区间应该为 $[o u t [4], i n [3]]$ ，也就是 $[6, 10]$ ，对应的编号为 $4 、 5 、 5 、 2 、 3$ ，我们发现 $5$ 出现了两次，出现了两次的点其实就相当于它实际上并不位于链上。所以在树上莫队中，判断一个点到底在不在区间内是看它出现的次数等于 $1$ 还是等于 $2$ ，我们可以用异或来实现这个判断。有眼见的同学可能发现了，你这编号里面没有 $1$ 啊，确实，当 $L C A (u, v) = k$ ， $k$ 不是 $u 、 v$ 时，我们计算出的区间其实时缺少了节点 $k$ 的，所以这里需要特判。至此树上莫队的关键知识点就讲完辣。代码中的一些细节问题就不赘述了。给一道模板题和实现代码。
https://vjudge.net/problem/SPOJ-COT2

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;

const int maxn=1e5+5;

int n,m,res,tot,num;
int a[maxn],b[maxn],pos[maxn],cnt[maxn];//莫队需要用到的
int oula[maxn],in[maxn],out[maxn]; //欧拉序需要用到的
int head[maxn],deep[maxn],f[maxn][17],bs[17];//LCA需要用到的
int ans[maxn];//记录结果
bool vis[maxn];//判断一个点是否在路径上

struct node
{
    int l,r,idx,lca=0;
    bool operator <(const node &a)const
    {
        if(pos[l]==pos[a.l])
            return r<a.r;
        return pos[l]<pos[a.l];
    }
}q[maxn];

struct Edge
{
    int to,nxt;
}edge[maxn];

inline void addedge(int u,int v)
{
    edge[++tot].to=v,edge[tot].nxt=head[u],head[u]=tot;
}

void dfs(int u,int fa)//LCA的预处理
{
    oula[++num]=u,in[u]=num;
    deep[u]=deep[fa]+1;
    f[u][0]=fa;
    for(int i=1;i<17;i++)
        f[u][i]=f[f[u][i-1]][i-1];
    int v;
    for(int i=head[u];i;i=edge[i].nxt)
    {
        v=edge[i].to;
        if(v!=fa)
            dfs(v,u);
    }
    oula[++num]=u,out[u]=num;
}

inline int skip(int x,int level)//求LCA要用到
{
    for(int i=0;i<17;i++)
        if(level&bs[i])
            x=f[x][i];
    return x;
}

inline int LCA(int u,int v)//求LCA
{
    if(deep[u]<deep[v])
        swap(u,v);
    u=skip(u,deep[u]-deep[v]);
    if(u==v)
        return u;
    for(int i=16;i>=0;i--)
        if(f[u][i]!=f[v][i])
            u=f[u][i],v=f[v][i];
    return f[u][0];
}

inline void work(int idx)//传进来的是节点编号
{
    vis[idx]^=1;
    if(vis[idx])//第一次出现的点 计算贡献
    {
        if(++cnt[a[idx]]==1)
            ++res;
    }
    else//第二次出现的点 说明这个点不在路径上
    {
        if(--cnt[a[idx]]==0)
            --res;
    }
}

int main()
{
    for(int i=0;i<17;i++)
        bs[i]=1<<i;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]),b[i]=a[i];
    sort(b+1,b+1+n);
    int nn=unique(b+1,b+1+n)-b-1;//因为颜色的范围比较大 所以需要离散化
    for(int i=1;i<=n;i++)
        a[i]=lower_bound(b+1,b+1+nn,a[i])-b;
    int u,v;
    for(int i=1;i<n;i++)   //连边
    {
        scanf("%d%d",&u,&v);
        addedge(u,v),addedge(v,u);
    }
    dfs(1,0);   //dfs做LCA的预处理 同时把欧拉序列求出来
    int dis=sqrt(num); //区间为[1,num]
    for(int i=1;i<=num;i++)
        pos[i]=(i-1)/dis+1;//莫队排序要用到
    int lca;
    for(int i=0;i<m;i++)
    {
        scanf("%d%d",&q[i].l,&q[i].r);
        lca=LCA(q[i].l,q[i].r);
        if(in[q[i].l]>in[q[i].r])//保证in[l]>=in[r] 这样计算得到的区间才合理哦
            swap(q[i].l,q[i].r);
        if(q[i].l==lca) //简单的情况
            q[i].l=in[q[i].l],q[i].r=in[q[i].r];
        else    //复杂的情况
            q[i].l=out[q[i].l],q[i].r=in[q[i].r],q[i].lca=lca;
        q[i].idx=i;
    }
    sort(q,q+m);//询问离线
    int l=1,r=0;
    for(int i=0;i<m;i++)
    {
        while(l<q[i].l)
            work(oula[l++]);
        while(l>q[i].l)
            work(oula[--l]);
        while(r<q[i].r)
            work(oula[++r]);
        while(r>q[i].r)
            work(oula[r--]);
        if(q[i].lca)  //特判掉LCA
            work(q[i].lca);
        ans[q[i].idx]=res;
        if(q[i].lca)  //要把LCA的贡献消除
            work(q[i].lca);
    }
    for(int i=0;i<m;i++)
        printf("%d\n",ans[i]);
    return 0;
}

对子树进行询问

子树问题，可以通过 $d f s$ 序将其转换为一段区间，我们还以上面的图为例：
在这里插入图片描述

1	2	3	4	5	6	7	8	9
1	2	4	7	5	3	6	8	9
9	4	2	1	1	4	3	2	1

第一行是 $d f s$ 序的编号，第二行是对应的节点编号，第三行是 $s i z [u]$ ，表示以 $u$ 为根的子树的节点个数，不难发现以 $u$ 为根的子树的区间可以表示为 $[d f n [u], d f n [u] + s i z [u] - 1]$ 。当然你也可以仿照欧拉序的方法，分别用 $i n [u] 、 o u t [u]$ 记录对应的区间，在 $d f s$ 退出时不递增编号就好了。其他的操作就和普通莫队差不多了。给一道模板题和代码。
https://vjudge.net/problem/CodeForces-375D

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;

const int maxn=1e5+5;

int n,m,tot,num;
int a[maxn],pos[maxn],cnt[maxn],cnk[maxn],ans[maxn];
int head[maxn],dfn[maxn],siz[maxn],id[maxn];

struct node
{
    int l,r,k,idx;
    bool operator <(const node &a)const
    {
        if(pos[l]==pos[a.l])
            return r<a.r;
        return pos[l]<pos[a.l];
    }
}q[maxn];

struct Edge
{
    int to,nxt;
}edge[maxn<<1];

inline void addedge(int u,int v)
{
    edge[++tot].to=v,edge[tot].nxt=head[u],head[u]=tot;
}

void dfs(int u,int fa)
{
    dfn[u]=++num,id[num]=u,siz[u]=1;
    int v;
    for(int i=head[u];i;i=edge[i].nxt)
    {
        v=edge[i].to;
        if(v==fa)
            continue;
        dfs(v,u);
        siz[u]+=siz[v];
    }
}

inline void add(int idx)
{
    ++cnt[a[idx]];
    ++cnk[cnt[a[idx]]];
}

inline void sub(int idx)
{
    --cnk[cnt[a[idx]]];
    --cnt[a[idx]];
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    int u,v;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        addedge(u,v),addedge(v,u);
    }
    int dis=sqrt(n);
    for(int i=1;i<=n;i++)
        pos[i]=(i-1)/dis+1;
    dfs(1,0);
    for(int i=0;i<m;i++)
    {
        scanf("%d%d",&q[i].l,&q[i].k);
        q[i].r=dfn[q[i].l]+siz[q[i].l]-1;
        q[i].l=dfn[q[i].l];
        q[i].idx=i;
    }
    sort(q,q+m);//询问离线
    int l=1,r=0;
    for(int i=0;i<m;i++)
    {
        while(l<q[i].l)
            sub(id[l++]);
        while(l>q[i].l)
            add(id[--l]);
        while(r<q[i].r)
            add(id[++r]);
        while(r>q[i].r)
            sub(id[r--]);
        ans[q[i].idx]=cnk[q[i].k];
    }
    for(int i=0;i<m;i++)
        printf("%d\n",ans[i]);
    return 0;
}