#前缀和
问题描述:
小蓝和小桥正在玩一个游戏,他们有一个长度为 n 的序列,其中既有偶数也有奇数,且偶数和奇数的数量相等。
小蓝有一些零花钱,他可以用这些钱来做一个特殊的操作:他在序列中选取一个位置,然后在这个位置上将序列分成两段,要求每一段中偶数和奇数的数量都相等。小蓝想要用他的零花钱尽可能多地进行这个操作,但每次操作都需要花费代价。具体而言,每次选取的位置可以看成是对序列进行切割,切割需要花费的代价为切割两端的元素的差的绝对值。小蓝想知道,在他的预算范围内,最多能进行多少次操作。
请你帮助小蓝计算最多可以进行的操作次数。
输入格式:
第一行包含两个整数 n 和 B( 2 ≤ n ≤ 100,1 ≤ B ≤ 100 ),表示序列的长度和小蓝拥有的零花钱数。
第二行包含 n 个整数 a1,a2,...,an( 1 ≤ ai ≤ 100 ),表示给定的序列(保证在这 n 个元素中奇数的个数等于偶数的个数)。
输出格式:
输出一个整数,表示在小蓝的预算范围内,能够进行的最多操作次数。、
样例输入:
6 3
1 2 3 4 5 6
样例输出:
2
题解:
#include <bits/stdc++.h>
using namespace std;
const int N = 200;
int a[N], pre[N], b[N];
int main()
{
ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
int n, B;
cin >> n >> B;
for (int i = 1; i <= n; i++) {
cin >> a[i];
pre[i] = pre[i - 1] + (a[i] & 1 ? 1 : -1);
}
int ans = 0;
for (int i = 1; i < n; i++) {
if (pre[i] == 0 && (max(a[i], a[i + 1]) - min(a[i], a[i + 1]) <= B)) {
b[ans++] = max(a[i], a[i + 1]) - min(a[i], a[i + 1]);
}
}
sort(b, b + ans);
int j = 0, sum = 0;
for (int i = 0; i < ans; i++) {
sum += b[i];
if (sum <= B)
j++;
}
cout << j << endl;
return 0;
}
总结:
在题解代码中,通过 a[i] & 1 ? 1 : -1 将奇数表示为 1 ,偶数表示为 -1 。
用一个前缀和数组pre [i] 存储是奇数还是偶数,如果前缀和数组为0,即 pre[i] == 0,表示奇数和偶数的数量相等,可以进行分割 。
通过 max(a[i], a[i + 1]) - min(a[i], a[i + 1]) <= B 判断小蓝的零花钱是否可以满足。
通过一个 数组b 存储小蓝所需的代价。
通过 sort(b, b + ans) 将代价排序 ,直到找到满足条件的最多次数。