蓝桥杯专题之前缀和篇

胃口很大的一条小蛇仔

已于 2022-10-10 11:32:35 修改

阅读量445

点赞数

分类专栏：蓝桥杯文章标签：算法

于 2022-04-04 11:50:56 首次发布

本文链接：https://blog.csdn.net/qq_53133234/article/details/123907569

版权

蓝桥杯专栏收录该内容

69 篇文章 25 订阅

订阅专栏

题目列表：

2014年：小朋友排队

2017年：k倍区间

二位前缀和

1.小朋友排队

问题描述
　　n 个小朋友站成一排。现在要把他们按身高从低到高的顺序排列，但是每次只能交换位置相邻的两个小朋友。

　　每个小朋友都有一个不高兴的程度。开始的时候，所有小朋友的不高兴程度都是0。

　　如果某个小朋友第一次被要求交换，则他的不高兴程度增加1，如果第二次要求他交换，则他的不高兴程度增加2（即不高兴程度为3），依次类推。当要求某个小朋友第k次交换时，他的不高兴程度增加k。

　　请问，要让所有小朋友按从低到高排队，他们的不高兴程度之和最小是多少。

　　如果有两个小朋友身高一样，则他们谁站在谁前面是没有关系的。
输入格式
　　输入的第一行包含一个整数n，表示小朋友的个数。
　　第二行包含 n 个整数 H1 H2 … Hn，分别表示每个小朋友的身高。
输出格式
　　输出一行，包含一个整数，表示小朋友的不高兴程度和的最小值。
样例输入
3
3 2 1
样例输出
9
样例说明
　　首先交换身高为3和2的小朋友，再交换身高为3和1的小朋友，再交换身高为2和1的小朋友，每个小朋友的不高兴程度都是3，总和为9。
数据规模和约定
　　对于10%的数据， 1<=n<=10；
　　对于30%的数据， 1<=n<=1000；
　　对于50%的数据， 1<=n<=10000；
　　对于100%的数据，1<=n<=100000，0<=Hi<=1000000。

分析：

这道题其实就是求逆序对，正一遍求逆序对再逆过来再求一遍逆序对，两遍地结果相加正好是每个小朋友需要移动的次数

例：

3 2 1

正： 0 1 2

反： 2 1 0

= 2 2 2

他们的不高兴程度就是3*（1+2）= 9

代码：

#include<iostream>
#include<stdio.h>
using namespace std;
const int N = 1e6+10;
int n;
int a[N];
long long sum[N];
int main(){
	cin >> n;
	for(int i = 1;i <= n;i++){
		 scanf("%d",&a[i]);
	}
	for(int i = 1;i <= n;i++){
		for(int j = i-1;j >= 1;j--){
			if(a[j] > a[i]){
				sum[i]++;
			}
		}
	}
	for(int i = n;i >= 1;i--){
		for(int j = i+1;j <= n;j++){
			if(a[j] < a[i]){
				sum[i]++;
			}
		}
	}
	long long ans = 0;
	for(int i = 1;i <= n;i++){
		ans += ((1+sum[i])*sum[i])/2;
	}
	cout << ans;
	return 0;
}

但是这样写只能过60%的数据，想要100%的通过，我们需要借助一个求逆序对的工具，那就是树状数组（如果不知道是怎么实现的，就去读我的另一篇博客是专门讲这道题的）

#include<iostream>
#include<cstring>
using namespace std;
const int MAX_N = 1e5+10;
int n;
int h[MAX_N]; 
long long cnt[MAX_N];//记录每个人的交换次数 
const int MaxH = 1e6+10; //最大的数 
long long c[MaxH];//树状数组 
int lowbit(int x){
	return x&(-x);
} 

long long query(int x){//求x的前缀和(即c[1]~c[x]的和) 
	long long res = 0;
	while(x>0){
		res+=c[x];
		x-=lowbit(x);
	}
	return res;
}

void update(int x,int v){//对c[x]进行修改
	while(x<=MaxH){
		c[x]+=v;
		x+=lowbit(x);
	} 
}

int main(){
	cin >> n;
	for(int i=0;i<n;i++){
		cin >> h[i];
	}
	
	for(int i=0;i<n;i++){//正序遍历
		update(h[i]+1,1);
		cnt[i] += i+1-query(h[i]+1);
	}
	memset(c,0,sizeof(c));//重置树状数组 
	
	for(int j=n-1;j>=0;j--){//逆序遍历 
		update(h[j]+1,1);
		cnt[j] += query(h[j]);
	}

	long long ans = 0;
	for(int i=0;i<n;i++){
		ans += (cnt[i]*(cnt[i]+1))/2;
	}
	cout << ans;
	return 0;
}

2.k倍区间

题目描述

给定一个长度为N的数列，A1, A2, … AN，如果其中一段连续的子序列Ai, Ai+1, … Aj(i <= j)之和是K的倍数，我们就称这个区间[i, j]是K倍区间。
你能求出数列中总共有多少个K倍区间吗？
输入
第一行包含两个整数N和K。(1 <= N, K <= 100000)
以下N行每行包含一个整数Ai。(1 <= Ai <= 100000)
输出
输出一个整数，代表K倍区间的数目。
样例输入
5 2
1
2
3
4
5
样例输出
6

分析：

同余定理：

如果a，b除以c的余数相同，就称a，b对于除数c来说是同余的，且有a与b的差能除c整除．（a，b，c均为自然数）

根据同余定理，我们可以知道如果a与b的差能被c整除，那么a和b就是同余的

那我们判断(sum[j]-sum[i-1])能不能被K整除，只需要判断sum[j]和sum[i-1]是不是同余的不就行了！

从上面我们知道如果任意两个前缀和同余，满足条件的区间数就+1，如果有m个前缀和同余，那么满足条件的区间数就有 $C_{m}^{2}\textrm{}$ ，即m*(m-1)/2个

所以我们只要统计同余0的前缀和有多少个，同余1的前缀和有多少个，同余2的前缀和有多少个……同余K-1的前缀和的有多少个，那么满足条件的区间的总个数就是： $\sum_{0}^{k-1} \frac{m*(m+1)}{2}$

代码：

#include<iostream>
using namespace std;
int N,K;
const int MAX_N = 1e5+10;
long long sum[MAX_N];
long long cnt[MAX_N];
int main(){
	cin >> N >> K;
	cnt[0] = 1;//千万别忘记sum[0] = 0这种情况
	for(int i = 1;i <= N;i++){
		cin >> sum[i];
		sum[i] = (sum[i-1] + sum[i]) % K;
		cnt[sum[i]]++;
	}
	long long ans = 0;
	for(int i = 0;i < K;i++){
		ans += cnt[i]*(cnt[i]-1)/2;
	}
	cout << ans;
	return 0;
}

二维前缀和：

（1）计算前 i 行 j 列的前缀和

s[i][j] = s[i-1][j] + s[i][j-1] - s[i-1][j-1] + a[i][j]（这里a[i][j]也就是s[i][j]，因为s[i][j]还没有更新，保存的实际上还是a[i][j])

二维前缀和转为一维前缀和

s[i][j] += s[i-1][j] (s[i][j]表示每一列的前缀和)

（2）计算任意区间的前缀和（（x1,y1）~（x2,y2））

s[x2][y2] - s[x2][y1-1] - s[x1-1][y2] + s[x1-1][y1-1]

代码：

for(int i = 1;i <= n;i++){
	for(int j = 1;j <= m;j++){
		s[i][j] = s[i-1][j] + s[i][j-1] - s[i-1][j-1] + s[i][j]; 
	}
}

for(int i = 1;i <= n;i++){
    for(int j = 1;j <= m;j++){
        s[i][j] += s[i-1][j];
    }
}


cin >> x1 >> y1 >> x2 >> y2;
cout << s[x2][y2] - s[x2][y1-1] - s[x1-1][y2] + s[x1-1][y1-1] << endl;