7 差分实现求区间插入和

钊气蓬勃.

已于 2022-03-19 17:36:42 修改

阅读量658

点赞数 2

文章标签：蓝桥杯算法 c语言

于 2021-12-29 09:21:14 首次发布

本文链接：https://blog.csdn.net/m0_63729880/article/details/122208475

版权

本文介绍了差分和前缀和的概念，并通过实例展示了它们在一维和二维数组中的应用。一维差分用于更新数组元素，二维差分则用于处理矩阵的前缀和问题。同时，还提供了利用差分和前缀和解决实际问题的两个例子：干草堆问题和增减序列问题。最后，以最高的牛问题为例，展示了如何在处理高度变化时使用差分和前缀和。

摘要由CSDN通过智能技术生成

一维差分

/*
学acwing的算法基础课学来的，喜欢的话多多支持呀。
*/
//一维差分，意思为an=bn的前缀和,an=bn+bn-1+bn-2...+b1
#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int a[N],b[N];//定义一个差分数组bn
void insert(int l,int r,int c)//定义一个插入区间函数
{
    b[l]+=c;
    b[r+1]-=c;
}
int main()
{
    int n,k;
    scanf("%d%d",&n,&k);//n个元素，k组插入
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);//读入an
     for(int i=1;i<=n;i++) insert(i,i,a[i]);//将an插入bn中，bn为an的差分
    while(k--)
    {
        int l,r,c;
        scanf("%d%d%d",&l,&r,&c);//读入插入区间，和插入的数
        insert(l,r,c);//插入bn数组中
    }
    for(int i=1;i<=n;i++) b[i]+=b[i-1];//求一次bn的前缀和，这样bn就是an了
      for(int i=1;i<=n;i++)  printf("%d ",b[i]);//输出an
    return 0;
}
/*到达胜利之前无法回头*/

二维差分

/*
学acwing的算法基础课学来的，喜欢的话多多支持呀。
*/
//二维差分，意思为矩阵的Anm等于Bnm的前缀和
#include<bits/stdc++.h>
using namespace std;
const int N=1e3+10;
int a[N][N],b[N][N];
void insert(int x1,int y1,int x2,int y2,int c)//定义一个插入子矩阵区间函数
{
    b[x1][y1]+=c;
    b[x2+1][y1]-=c;
    b[x1][y2+1]-=c;
    b[x2+1][y2+1]+=c;
}
int main()
{
    int n,m,k;
    scanf("%d%d%d",&n,&m,&k);//n*m矩阵，k组插入
    for(int i=1;i<=n;i++)
         for(int j=1;j<=m;j++)
          scanf("%d",&a[i][j]);//读入anm
     for(int i=1;i<=n;i++)
          for(int j=1;j<=m;j++)
        insert(i,j,i,j,a[i][j]);//将anm插入bnm中，bnm为anm的差分
    while(k--)
    {
        int x1,y1,x2,y2,c;
        scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&c);//读入插入区间，和插入的数
        insert(x1,y1,x2,y2,c);//插入bnm数组中
    }
    for(int i=1;i<=n;i++)
         for(int j=1;j<=m;j++)
            b[i][j]+=b[i-1][j]+b[i][j-1]-b[i-1][j-1];//求bnm前缀和，求法跟上一节求前缀和一样，求出就是anm
   for(int i=1;i<=n;i++)
       {
            for(int j=1;j<=m;j++) printf("%d ",b[i][j]);//输出anm
            printf("\n");
       } 
    return 0;
}
/*到达胜利之前无法回头*/

补题中发现的差分

干草堆差分+前缀和

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int b[N];
void insert(int l,int r,int c)//差分
{
    b[l]+=c;
    b[r+1]-=c;
}
int main()
{
    int n,k;
    scanf("%d%d",&n,&k);
    while(k--)
    {
        int l,r;
        scanf("%d%d",&l,&r);
        insert(l,r,1);//在l,r区间插入1
    }
    for(int i=1;i<=n;i++) b[i]+=b[i-1];//求一次前缀和
    sort(b+1,b+n+1);
 printf("%d\n",b[n/2+1]);
    return 0;
}

增减序列

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+10;
int a[N];
int main()
{
    int n;
   cin>>n;
   for(int i=1;i<=n;i++) scanf("%d",&a[i]);
   for(int i=n;i>1;i--) a[i]-=a[i-1];//差分a[i];
   ll pos=0,neg=0;//pos存正数，neg存负数
   for(int i=2;i<=n;i++)
     if(a[i]>0) pos+=a[i];
     else neg-=a[i];
     cout<<min(pos,neg)+abs(pos-neg)<<endl;//输出操作最小次数
     cout<<abs(pos-neg)+1<<endl;//输出变成的种数
    return 0;
}

最高的牛

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e4+10;
int h[N];
int main()
{
    int n,p,height,m;
   cin>>n>>p>>height>>m;
   h[1]=height;//定义每个点为最高高度
   set<pair<int,int>> existed;//用来判断点是否出现过
   for(int i=0;i<m;i++)
   {
       int a,b;
       cin>>a>>b;
       if(a>b) swap(a,b);
       if(!existed.count({a,b}))
        {
            existed.insert({a,b});
            h[a+1]--,h[b]++;//差分
        }
   }
   for(int i=1;i<=n;i++)//求一下原数组，前缀和
   {
       h[i]+=h[i-1];
       cout<<h[i]<<endl;
   }
    return 0;
}