莫队算法

49 篇文章 0 订阅
24 篇文章 0 订阅

莫涛大神发明的似乎可解一切区间问题的算法,本蒟蒻觉得似乎其本质是分块优化再加上多关键字排序。分块大法好
拿题说事

袜子

这是道很好拿来理解莫队的题

题目大意

进行区间询问[l,r],输出该区间内随机抽两次抽到相同颜色袜子的概率。

分析

当你确定区间[l,r]的答案时(分子为各种颜色的平方和,分母为(l-r)^2),可借其进行分析,当区间变为[l-1,r]时,设l块颜色为co(原来的贡献为x),则分母变为(l-r-1)^2,分母减去x^2+(x-1)^2,可总结出当该变量为1时,可O(1)进行修改,然而这还是不够的,毕竟在最坏条件下还是O(n^2),这时就需要神奇的排序了,先将序列分为根号n段,每块大为k,在排序时,若在同一段则按r排序,否则按l排序,即将l作为第一关键字,r作为第二关键字进行排序,对于l来说,最多经过2*k即可完成目标,复杂度为:O(m*k),对于r来说,只有在同一段里时才有序,而对于每个块来说,最多移动n次,一共n/k个块,故复杂度为O(n*n/k),所以总复杂度为O(n*sqrt(n)),实际为O(玄学)反正能过就行
上代码

#include<bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof(a))
#define ll long long 
using namespace std;
struct date{
    int l,r,ID;
    ll A,B;
}q[50003];
ll S(ll x){ return x*x;}
ll GCD(ll a,ll b){
    return b==0?a:GCD(b,a%b);
}
long long n,m,col[50003],unit,x[50003];
ll sum[50003],ans;
bool cmp(date a,date b){
    return x[a.l]==x[b.l]?a.r<b.r:a.l<b.l;
}
bool CMP(date a,date b){
    return a.ID<b.ID;
}
void cl(int x,int add){
    ans-=S(sum[col[x]]);
    sum[col[x]]+=add;
    ans+=S(sum[col[x]]);
}
int main()
{
    scanf("%d%d",&n,&m);
    unit=sqrt(n);
    for(int i=1;i<=n;i++) scanf("%d",&col[i]),x[i]=i/unit+1;;
    for(int i=1;i<=m;i++) scanf("%d%d",&q[i].l,&q[i].r),q[i].ID=i;
    sort(q+1,q+m+1,cmp);
    int l=1,r=0;
    for(int i=1;i<=m;i++)
    {
        while(l<q[i].l) cl(l,-1),l++;
        while(l>q[i].l) cl(l-1,1),l--;
        while(r<q[i].r) cl(r+1,1),r++;
        while(r>q[i].r) cl(r,-1),r--;
        if(q[i].l==q[i].r){
            q[i].A=0,q[i].B=1;
            continue;
        }
        q[i].A=ans-q[i].r+q[i].l-1;
        q[i].B=1LL*(q[i].r-q[i].l+1)*(q[i].r-q[i].l);
        ll gcd=GCD(q[i].A,q[i].B);
        q[i].A/=gcd;
        q[i].B/=gcd;
    }
    sort(q+1,q+m+1,CMP);
    for(int i=1;i<=m;i++) printf("%lld/%lld\n",q[i].A,q[i].B);
    return 0;
}
莫队算法是一种基于分块的算法,用于解决一些静态区间查询问题,时间复杂度为 $O(n\sqrt{n})$。以下是一个基于Python的莫队算法的示例代码: ```python import math # 定义块的大小 BLOCK_SIZE = 0 # 初始化块的大小 def init_block_size(n): global BLOCK_SIZE BLOCK_SIZE = int(math.sqrt(n)) # 定义查询操作 def query(left, right): pass # 在这里写查询操作的代码 # 定义添加操作 def add(x): pass # 在这里写添加操作的代码 # 定义删除操作 def remove(x): pass # 在这里写删除操作的代码 # 定义莫队算法 def mo_algorithm(n, q, queries): init_block_size(n) queries.sort(key=lambda x: (x[0] // BLOCK_SIZE, x[1])) left, right = 0, -1 for query in queries: while left > query[0]: left -= 1 add(left) while right < query[1]: right += 1 add(right) while left < query[0]: remove(left) left += 1 while right > query[1]: remove(right) right -= 1 query(query[0], query[1]) ``` 在这段代码中,我们首先定义了一个全局变量 `BLOCK_SIZE`,用于表示块的大小。接着,我们定义了三个操作函数 `query()`、`add()` 和 `remove()`,分别用于查询、添加和删除元素。在 `mo_algorithm()` 函数中,我们首先调用 `init_block_size()` 函数初始化块的大小,然后将查询操作按照块的大小和右端点排序,接着使用双指针维护当前查询区间的左右端点,每次移动指针时调用 `add()` 和 `remove()` 函数更新块的状态,最后调用 `query()` 函数进行查询操作。 请注意,这段代码只是一个示例,具体的 `query()`、`add()` 和 `remove()` 函数的实现取决于具体的问题。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值