c++算法之前缀和

至尊宝加油

已于 2024-01-20 17:21:48 修改

阅读量672

点赞数 9

文章标签： c++ 算法

于 2024-01-20 17:19:38 首次发布

本文链接：https://blog.csdn.net/weixin_63201442/article/details/135717847

版权

前缀和原理和特点

prefix表示前缀和，前缀和由一个用户输入的数组生成。

prefix[x] = a[1]+a[2]+...a[i]

同理：prefix[x] = prefix[x-1]+a[x]

求一段区间的和：sum(l,r) = prefix[r]-prefix[l-1]

但是注意：prefix是一种预处理算法，只适用于a数组为静态数组的情况，即a数组中的元素在区间和查询过程中不会进行修改。

如果需要实现“先区间修改，再区间查询”可以使用差分数组，如果需要“一边修改一边查询”需要使用树状数组或线段树等数据结构。

实现前缀和

for(int i = 1;i <= n;++ i)
{
    prefix[i] = prefix[i-1]+a[i];
}

例题

蓝桥oj3382 区间次方和

问题描述

给定一个长度为 nn 的整数数组 aa 以及 mm 个查询。

每个查询包含三个整数 l,r,kl,r,k 表示询问 l∼rl∼r 之间所有元素的 kk 次方和。

请对每个查询输出一个答案，答案对 109+7109+7 取模。

输入格式

第一行输入两个整数 n,mn,m 其含义如上所述。

第二行输入 nn 个整数 a[1],a[2],...,a[n]a[1],a[2],...,a[n]。

接下来 mm 行，每行输入三个整数 l,r,kl,r,k 表示一个查询。

输出格式

输出 mm 行，每行一个整数，表示查询的答案对 109+7109+7 取模的结果。

样例输入

样例输出

14
99
12

解

#include <iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 1e5 + 9;
using ll = long long;
const ll p = 1e9 + 7;
ll a[6][N], prefix[6][N];


int main()
{
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	int n, m; cin >> n >> m;
	for (int i = 1; i <= n; i++)cin >> a[1][i];

	for (int i = 2; i <= 5; ++i)
		for (int j = 1; j <= n; ++j)
			a[i][j] = a[i - 1][j] * a[1][j];

	for (int i = 1; i <= 5; ++i)
	{
		for (int j = 1; j <= n; j++)
			prefix[i][j] = (prefix[i][j - 1] + a[i][j]);
	}
		while (m--)
		{
			int l, r, k; cin >> l >> r >> k;
			cout << (prefix[k][r] - prefix[k][l-1]) % p << endl;
		}

	
	// 请在此输入您的代码
	return 0;
}

蓝桥oj3419小郑的蓝桥平衡串

问题描述

平衡串指的是一个字符串，其中包含两种不同字符，并且这两种字符的数量相等。

例如，ababab 和 aababb 都是平衡串，因为每种字符各有三个，而 abaab和 aaaab 都不是平衡串，因为它们的字符数量不相等。

平衡串在密码学和计算机科学中具有重要应用，比如可以用于构造哈希函数或者解决一些数学问题。

小郑拿到一个只包含 L、Q 的字符串，他的任务就是找到最长平衡串，且满足平衡串的要求，即保证子串中 L、Q 的数量相等。

输入格式

输入一行字符串，保证字符串中只包含字符 L、Q。

输出格式

输出一个整数，为输入字符串中最长平衡串的长度。

样例输入

LQLL

样例输出

解

#include <iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 1010;
char s[N];

int prefix[N];

int main()
{
	cin >> s + 1;
	int n = strlen(s + 1);
	for (int i = 1; i <= n; ++i)prefix[i] = prefix[i - 1] + (s[i] == 'L' ? 1 : -1);

	int ans = 0;
	for (int i = 1; i <= n; i++)
	{
		for (int j = i; j <= n; j++)
		{
			if (prefix[j] - prefix[i - 1] == 0)ans = max(ans, j - i + 1);
		}
	}
	cout << ans << endl;

	return 0;
}