前缀和与差分

1.一维前缀和

//知识点
a[1],a[2],...a[n]
s[i]=a[1]+a[2]+...+a[i]
a[3]+a[4]+....+a[15]=s[15]-s[2]

例题一·:
输入样例:

【样例13
1 2 3
【样例24
1 3 1 1
【样例38
27 23 76 2 3 5 62 52

输出样例:

【样例10
【样例22
【样例32

题目描述:
如样例所示,给出一串数字,让后将其左右分成两部分,使其和之差达到最小。
代码如下:

#include<stdio.h>
int main()
{
    int n;
    int arr[101],brr[101],sum=0,min=101;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%d",&arr[i]);
        sum=sum+arr[i];
        brr[i]=sum;//建立前缀和数组
    }
    for(int i=0;i<n-1;i++)
    {
        if(brr[i]>brr[n-1]-brr[i])
        {
            if(2*brr[i]-brr[n-1]<min)
                min=2*brr[i]-brr[n-1];
        }
        else
        {
            if(brr[n-1]-2*brr[i]<min)
                min=brr[n-1]-2*brr[i];
        }
    }
    printf("%d",min);
    return 0;
}

例题二:
输入样例:

5
WEEWW

输出样例:

1

题目描述:
如样例所示,在字符串中,选择其中一个字符作为‘领导’,然后其他字符都要面向他,W代表向左看,E代表向右看,统计将哪个字符作为领导时,需要转向的字符最少。
代码如下:

#include<stdio.h>
int main()
{
    char arr[1000001];
    int n,count=0,max=0;
    scanf("%d",&n);
    scanf("%s",arr);
    for(int i=1;i<n;i++)
    {
        if(arr[i]=='W')
            count++;
    }
    max=count;
    for(int i=1;i<n;i++)//此处间接利用前缀和的思想,当前一个字符的权值确定时,要想确定当前字符的权值,只需看前一个字符与当前字符,无需通过全局判断。
    {
        if(arr[i-1]=='E')
            count++;
        if(arr[i]=='W')
            count--;
        if(count>max)
            max=count;
    }
    printf("%d",n-max-1);
    return 0;
}

2.二维前缀和

求二维矩阵固定长度的累加和的最大值
代码如下:

//二维前缀和
#include<bits/stdc++.h>
using namespace std;
int main()
{
    int arr[100][100]={0};
    int a,b,R;//求边长为R的正方形中的权值和
    scanf("%d %d %d",&a,&b,&R);
    for(int i=1;i<=a;i++)
    {
        for(int j=1;j<=b;j++)
        {
            scanf("%d",&arr[i][j]);
        }
    }
    for(int i=1;i<=a;i++)//创建二维前缀和
    {
        for(int j=1;j<=b;j++)
        {
            arr[i][j]+=arr[i][j-1]+arr[i-1][j]-arr[i-1][j-1];
        }
    }
    int maxA=0;
    for(int i=R;i<=a;i++)//统计最大值,注意下标
    {
        for(int j=R;j<=b;j++)
        {
            maxA=max(maxA,arr[i][j]-arr[i-R][j]-arr[i][j-R]+arr[i-R][j-R]);
        }
    }
    printf("%d",maxA);
    return 0;
}

3.一维差分

//知识点
a[1],a[2],...,a[n]
b[i]=a[i]-a[i-1],b[1]=a[1]  b是a的差分序列,a是b的前缀和序列
证明后者:
a[i]=b[1]+b[2]+b[3]+...+b[i]
    =a[1]+a[2]-a[1]+a[3]-a[2]+...+a[i]-a[i-1]
    =a[i]

测试样例:

输入:
4
1 1 2 2
输出:
1
2

题目描述:
给定一个长度为n的数列,每次可以选择一个区间[l,r],使下标在这个区间内的数都执行加一或减一操作,求至少需要多少次才能使数列中的所有数都相等,并求出数列在最小次数的前提下可能的种数。
解题思路:

给区间[l,r]加上一个常数C == 对其差分序列:b[l]+=C,b[r+1]+=C
设i,j为差分序列的下标,我们可以对差分序列做以下四种操作:
1. 2<=i,j<=n(改变区间[i,j])
2. i=1,2<=j<=n(改变区间[1,j])
3. 2<=i<=n,j=n+1(改变区间[i,n])
4. i=1,j=n+1(改变整个区间,对本题来说没有实际意义)

代码如下:

//差分
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int n=10010;
int a[n];
int main()
{
    int n;
    cin>>n;
    for(int i=1; i<=n; i++)cin>>a[i];
    for(int i=n; i>1; i--)a[i]-=a[i-1];
    ll pos=0,neg=0;
    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;
}

4.二维差分(矩阵差分)

题目描述:
输入一个 n 行 m 列的整数矩阵,再输入 q 个操作,每个操作包含五个整数 x1,y1,x2,y2,c,其中 (x1,y1) 和 (x2,y2) 表示一个子矩阵的左上角坐标和右下角坐标。每个操作都要将选中的子矩阵中的每个元素的值加上 c。请你将进行完所有操作后的矩阵输出。
输入输出样例:

输入:
3 4 3
1 2 2 1
3 2 2 1
1 1 1 1
1 1 2 2 1
1 3 2 3 2
3 1 3 4 1
输出:
2 3 4 1
4 3 4 1
2 2 2 2

代码如下:

#include<iostream>
using namespace std;

const int N=1010;
int a[N][N],b[N][N];//b为差分数组
void insert(int x1,int y1,int x2,int y2,int c)
{
    b[x1][y1]=b[x1][y1]+c;
    b[x2+1][y1]=b[x2+1][y1]-c;
    b[x1][y2+1]=b[x1][y2+1]-c;
    b[x2+1][y2+1]=b[x2+1][y2+1]+c;
}
int main()
{
    int n,m,q;
    scanf("%d %d %d",&n,&m,&q);
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
            scanf("%d",&a[i][j]);
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
            b[i][j]=a[i][j]-a[i-1][j]-a[i][j-1]+a[i-1][j-1];//求差分数组
    }
    while(q--)
    {
        int x1,y1,x2,y2,c;
        scanf("%d %d %d %d %d",&x1,&y1,&x2,&y2,&c);
        insert(x1,y1,x2,y2,c);
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            b[i][j]=b[i][j]+b[i-1][j]+b[i][j-1]-b[i-1][j-1];
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
            printf("%d ",b[i][j]);
        printf("\n");
    }
    return 0;
}

5.绳索折叠(应用差分)

在这里插入图片描述

#include<iostream>
#include<algorithm>
using namespace std;

int main()
{
    int n,l,arr[10010],s[10010];
    scanf("%d %d",&n,&l);
    if(n==1)
        printf("0");
    else if(n==2)
        printf("1");
    else
    {
        for(int i=0; i<n; i++)
        {
            scanf("%d",&arr[i]);
        }
        sort(arr,arr+n);
        for(int i=0; i<n-1; i++)
        {
            s[i]=arr[i+1]-arr[i];
        }
        int count=0;
        for(int i=0; i<n-2; i++) //找点
        {
            int z=i,y=i+1,flag1=1;
            while(z>=0&&y<n-1)
            {
                if(s[z]!=s[y])
                {
                    flag1=0;
                    break;
                }
                z--;
                y++;
            }
            if(flag1==1)
                count++;
        }
        for(int i=0; i<n-3; i++) //找空
        {
            int z=i,y=i+2,flag2=1;
            while(z>=0&&y<n-1)
            {
                if(s[z]!=s[y])
                {
                    flag2=0;
                    break;
                }
                z--;
                y++;
            }
            if(flag2==1)
                count++;
        }
        printf("%d",count+2);
    }
    return 0;
}

6. 野牛与狼

在这里插入图片描述

#include<iostream>
#include<string.h>
#include<map>
using namespace std;
 
map<int,int>mp;
int s[1000010];
int main()
{
    int maxA=0;
    char arr[1000010];
    scanf("%s",arr);
    for(int i=0;i<strlen(arr);i++)
    {
        if(arr[i]=='1') s[i+1]=s[i]+1;
        else s[i+1]=s[i]-1;
    }
    for(int i=1;i<=strlen(arr);i++)
    {
        if(s[i]==0)
        {
            if(i>maxA) maxA=i;
        }
        else
        {
            if(mp[s[i]]==0) mp[s[i]]=i;
            else
            {
                if(i-mp[s[i]]>maxA) maxA=i-mp[s[i]];
            }
        }
    }
    printf("%d",maxA);
    return 0;
}

7. E. Counting Rectangles(CF)

在这里插入图片描述
在这里插入图片描述

链接:https://codeforces.com/contest/1722/problem/E
题意:给定n个矩形的长和宽,然后再给出q组询问,每组询问有两个举行的长和宽,求在两个举行范围之内的矩形的面积总和

#include<iostream>
#include<cstring>
using namespace std;
 
int t;
long long arr[1010][1010];
 
int main()
{
    cin>>t;
    int n,q;
    while(t--)
    {
        memset(arr,0,sizeof arr);
        scanf("%d %d",&n,&q);
        int h,w;
        //int maxh=0,maxw=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d %d",&h,&w);
            arr[h][w]+=h*w;
        }
        for(int i=1;i<=1000;i++)
        {
            for(int j=1;j<=1000;j++)
            {
                arr[i][j]+=arr[i-1][j]+arr[i][j-1]-arr[i-1][j-1];
            }
        }
        int hs,ws,hb,wb;
        while(q--)
        {
            scanf("%d %d %d %d",&hs,&ws,&hb,&wb);
            printf("%lld\n",arr[hb-1][wb-1]-arr[hs][wb-1]-arr[hb-1][ws]+arr[hs][ws]);
        }
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

勇敢nn

心动不如行动,感谢您的支持!

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值