分块练习题_分块练习题-CSDN博客

本文链接：https://blog.csdn.net/cheng__yu_/article/details/108148510

习题

243. 一个简单的整数问题2

链接：https://www.acwing.com/problem/content/244/
在这里插入图片描述

题意：维护区间修改，维护区间查询

思路：分块维护

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=1e5+10;

int n,m,B,a[maxn]; 
ll res[maxn],base[maxn];

ll query(int l,int r)
{
	int idl=l/B,idr=r/B;
	ll ans=0;
	if(idl==idr)
	{
		for(int i=l;i<=r;++i)
			ans+=a[i]+base[idl];
	}
	else
	{
		for(int i=l;i<(idl+1)*B;++i)
			ans+=a[i]+base[idl];
		for(int i=idl+1;i<=idr-1;++i)
			ans+=res[i];
		for(int i=idr*B;i<=r;++i)
			ans+=a[i]+base[idr];
	}
	return ans;
} 

void update(int l,int r,int x)
{
	int idl=l/B,idr=r/B;
	ll ans=0;
	if(idl==idr)
	{
		for(int i=l;i<=r;++i)
		{
			a[i]+=x;
			res[idl]+=x;
		}
	}
	else
	{
		for(int i=l;i<(idl+1)*B;++i)
		{
			a[i]+=x;
			res[idl]+=x;
		}
		for(int i=idl+1;i<=idr-1;++i)
		{
			base[i]+=x;
			res[i]+=B*x;
		}
		for(int i=idr*B;i<=r;++i)
		{
			a[i]+=x;
			res[idr]+=x;
		}
	}	
}

int main()
{
	scanf("%d%d",&n,&m);
	B=sqrt(n);
	for(int i=1;i<=n;++i)
	{
		scanf("%d",&a[i]);
		res[i/B]+=a[i];
	}
	char op[10];
	int l,r,x;
	while(m--)
	{
		scanf("%s",op);
		if(op[0]=='Q')
		{
			scanf("%d%d",&l,&r);
			printf("%lld\n",query(l,r));
		}
		else
		{
			scanf("%d%d%d",&l,&r,&x);
			update(l,r,x);
		}
	}
	return 0;
}

CF785E Anton and Permutation

链接：https://www.luogu.com.cn/problem/CF785E

题意：给定一个排列，每次交换两个位置，求逆序对

思路：分块，每次找在下标在（l ，r），值域在（x，y）上的数字个数。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=2e5+10;

int n,m,B;
int a[maxn];
vector<int> vec[maxn];

ll solve(int l,int r)
{
	if(l>=r) return 0;
	int x=a[l],y=a[r],f,k;
	int idl=l/B,idr=r/B;
	k=find(vec[idl].begin(),vec[idl].end(),x)-vec[idl].begin();
	vec[idl][k]=y;
	sort(vec[idl].begin(),vec[idl].end());
	
	k=find(vec[idr].begin(),vec[idr].end(),y)-vec[idr].begin();
	vec[idr][k]=x;
	sort(vec[idr].begin(),vec[idr].end());
	
	if(x<y) f=1;
	else f=-1,swap(x,y);
	
	ll ans=0;
	if(idl==idr)
	{
		for(int i=l+1;i<=r-1;++i)
			if(a[i]>x&&a[i]<y) ans++;
	}
	else
	{
		for(int i=l+1;i<idl*B+B;++i)		
			if(a[i]>x&&a[i]<y) ans++;
		for(int i=idl+1;i<=idr-1;++i)
			ans+=upper_bound(vec[i].begin(),vec[i].end(),y)-lower_bound(vec[i].begin(),vec[i].end(),x);
		for(int i=idr*B;i<=r-1;++i)
			if(a[i]>x&&a[i]<y) ans++;
	}
	swap(a[l],a[r]);
	return f*(2*ans+1);	
}

int main()
{
	scanf("%d%d",&n,&m);
	B=sqrt(n);
	for(int i=1;i<=n;++i)
	{
		a[i]=i;
		vec[i/B].push_back(i); 
	}
	int l,r;
	ll ans=0;
	while(m--)
	{
		scanf("%d%d",&l,&r);
		if(l>r) swap(l,r);
		ans+=solve(l,r);
		printf("%lld\n",ans);
	}
	return 0;
}

249. 蒲公英

链接：https://www.acwing.com/problem/content/description/251/

题意：给定一个序列 $a_1,a_2,\dots,a_n$ 。有 m 次询问，每次询问区间 $[l, r]$ 上的众数。有多个众数时，输出值较小的那个。强制在线。

思路：不是强制在线的话，可以用莫队来写。分块，一个区间的众数来自于，块间的众数和边界块内的数的贡献。

预处理出任意两个块之间众数。
对于暴力枚举边界块的 $a_i$ ，需要知道 $a_i$ 在 [idl+1,idr-1] 中出现的次数。这里可以用预处理前缀和的方式得出。处理出，每个块 $a_1$ 至 $a_n$ 出现的次数
时间复杂度在 $m\sqrt n$

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=4e4+5;

int n,m,a[maxn],b[maxn],B;
int d[210][210],cnt[maxn];
int pref[201][maxn];

void init()
{
    for(int i=1; i<=n; ++i)
    {
        if(i%B==0)
        {
            for(int j=1; j<=n; ++j)
                if(i/B-1>=0) pref[i/B][a[j]]=pref[i/B-1][a[j]];
        }
        pref[i/B][a[i]]++;
    }

    for(int l=0; l<=n; l+=B)
    {
        if(l==0) continue;
        memset(cnt,0,sizeof(cnt));
        int mx=0,tp=0;
        for(int r=l; r<=n; ++r)
        {
            cnt[a[r]]++;
            if(cnt[a[r]]>mx||cnt[a[r]]==mx&&a[r]<tp) mx=cnt[a[r]],tp=a[r];
            if((r+1)%B==0||r==n) d[l/B][r/B]=tp;
        }
    }
}

inline int calc(int idl,int idr,int x)
{
    if(idl>idr) return 0;
    if(idl<=0) return pref[idr][x];
    return pref[idr][x]-pref[idl-1][x];
}

int query(int l,int r)
{
    int idl=l/B,idr=r/B;
    int mx=0,tp=0;
    if(idl+1<=idr-1) tp=d[idl+1][idr-1];
    mx=calc(idl+1,idr-1,tp);
    if(idl==idr)
    {
        for(int i=l; i<=r; ++i) cnt[a[i]]++;
        for(int i=l; i<=r; ++i)
        {
            int num=cnt[a[i]];
            if(num>mx||num==mx&&a[i]<tp) mx=num,tp=a[i];
            cnt[a[i]]=0;
        }
    }
    else
    {
        for(int i=l; i<(idl+1)*B; ++i) cnt[a[i]]++;
        for(int i=idr*B; i<=r; ++i) cnt[a[i]]++;

        for(int i=l; i<(idl+1)*B; ++i)
        {
            int num=calc(idl+1,idr-1,a[i])+cnt[a[i]];
            if(num>mx||num==mx&&a[i]<tp) mx=num,tp=a[i];
            cnt[a[i]]=0;
        }
        for(int i=idr*B; i<=r; ++i)
        {
            int num=calc(idl+1,idr-1,a[i])+cnt[a[i]];
            if(num>mx||num==mx&&a[i]<tp) mx=num,tp=a[i];
            cnt[a[i]]=0;
        }
    }
    return b[tp];
}
int main()
{
    scanf("%d%d",&n,&m);
    B=sqrt(n);
    for(int i=1; i<=n; ++i) scanf("%d",&a[i]),b[i]=a[i];
    sort(b+1,b+1+n);
    for(int i=1; i<=n; ++i) a[i]=lower_bound(b+1,b+1+n,a[i])-b;
    init();
    memset(cnt,0,sizeof(cnt));
    int ans=0;
    while(m--)
    {
        int l,r;
        scanf("%d%d",&l,&r);
        l=(l+ans-1)%n+1;
        r=(r+ans-1)%n+1;
        if(l>r) swap(l,r);
        ans=query(l,r);
        printf("%d\n",ans);
    }
    return 0;
}

P3157 [CQOI2011]动态逆序对

链接：https://www.luogu.com.cn/problem/P3157

题意：给定 n 的排列，每次删除一个数。询问删除前的逆序对。

思路：其实就是询问一下删除的这个数，前面比自己大的数有多少、后面比自己小的数有多少。

分块，每个块开一个树状数组来维护，然后询问即可。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=1e5+5,inf=1e6;

int n,m,a[maxn],p[maxn],B;
int c[400][maxn],sz[maxn];
set<int> vec[400];

int lowbit(int x)
{
    return x&-x;
}
void add(int b,int x,int val)
{
    for(int i=x; i<=n; i+=lowbit(i))
        c[b][i]+=val;
}
int getsum(int b,int x)
{
    int res=0;
    for(int i=x; i>0; i-=lowbit(i))
        res+=c[b][i];
    return res;
}
int solve(int x)
{
    int pos=p[x];
    add(pos/B,a[pos],-1);
    a[pos]=inf;
    sz[pos/B]--;

    int ans1=0,ans2=0;
    if(pos>1)
    {
        int l=1,r=pos-1;
        int idl=l/B,idr=r/B;
        if(idl==idr)
        {
            for(int i=l; i<=r; ++i)
                if(a[i]!=inf&&a[i]>x) ans1++;
        }
        else
        {
            for(int i=l; i<(idl+1)*B; ++i)
                if(a[i]!=inf&&a[i]>x) ans1++;
            for(int i=idl+1; i<=idr-1; ++i)
                ans1+=sz[i]-getsum(i,x);

            for(int i=idr*B; i<=r; ++i)
                if(a[i]!=inf&&a[i]>x) ans1++;
        }
    }
    if(pos<n)
    {
        int l=pos+1,r=n;
        int idl=l/B,idr=r/B;
        if(idl==idr)
        {
            for(int i=l; i<=r; ++i)
                if(a[i]!=inf&&a[i]<x) ans2++;
        }
        else
        {
            for(int i=l; i<(idl+1)*B; ++i)
                if(a[i]!=inf&&a[i]<x) ans2++;
            for(int i=idl+1; i<=idr-1; ++i)
                ans2+=getsum(i,x);
            for(int i=idr*B; i<=r; ++i)
                if(a[i]!=inf&&a[i]<x) ans2++;
        }
    }
    return ans1+ans2;
}
int main()
{
    scanf("%d%d",&n,&m);
    B=sqrt(n)*5;
    for(int i=1; i<=n; ++i) scanf("%d",&a[i]),p[a[i]]=i;

    ll ans=0;
    for(int i=1; i<=n; ++i)
    {
        add(n/B+1,a[i],1);
        ans+=i-getsum(n/B+1,a[i]);
        add(i/B,a[i],1);
        sz[i/B]++;
    }
    while(m--)
    {
        printf("%lld\n",ans);
        int x;
        scanf("%d",&x);
        ans-=solve(x);
    }
    return 0;
}