中国做网站最好的公司一张简单的网页多少钱

0合并数列 - 蓝桥云课

问题描述

小明发现有很多方案可以把一个很大的正整数拆成若干正整数的和。他采取了其中两种方案，分别将他们列为两个数组 {a1, a2, ..., an} 和 {b1, b2, ..., bm}。两个数组的和相同。

定义一次合并操作可以将某数组内相邻的两个数合并为一个新数，新数的值是原来两个数的和。小明想通过若干次合并操作将两个数组变成一模一样，即 n = m 且对于任意下标 i 满足 a_i = b_i。请计算至少需要多少次合并操作可以完成小明的目标。

输入格式

输入共3行。

第一行为两个正整数 n, m。

第二行为 n 个由空格隔开的整数 a1, a2, ..., an。

第三行为 m 个由空格隔开的整数 b1, b2, ..., bm。

输出格式

输出共1行，一个整数。

样例输入

4 3
1 2 3 4
1 5 4

样例说明

只需要将 a2 和 a3 合并，数组 a 变为 {1, 5, 4}，即和 b 相同。

评测用例规模与约定

对于 20% 的数据，保证 n, m ≤ 10^3。

对于 100% 的数据，保证 n, m ≤ 10^5，0 < a_i, b_i ≤ 10^5。

思路：

暴力遍历，双指针

代码如下：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N = 1e5+10;
ll n,m;
ll a[N],b[N]; 
ll ans = 0;
int main()
{cin >> n >> m;for(ll i = 1 ; i <= n ; i++)cin >> a[i];for(ll i = 1 ; i <= m ; i++)cin >> b[i];ll l = 1,r = 1;while(l <= n && r <= m){if(a[l] == b[r]){l++;r++;}else if(a[l] < b[r]){a[l+1] += a[l];l++;ans++;}else if(a[l] > b[r]){b[r+1] += b[r];r++;ans++;}}cout << ans;return 0;
}

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N = 1e5+10;
ll n,m;
ll a[N],b[N]; 
ll ans = 0;int main() {cin >> n >> m;for (ll i = 1; i <= n; i++) cin >> a[i];for (ll i = 1; i <= m; i++) cin >> b[i];ll l = 1, r = 1;while (l <= n && r <= m) {  // 修正循环条件if (a[l] == b[r]) {l++;r++;} else if (a[l] < b[r]) {ll sum = a[l];l++;while (sum < b[r] && l <= n) { // 合并 a 数组的元素sum += a[l];l++;ans++;}r++; // 此时 sum >= b[r]} else { // a[l] > b[r]ll sum = b[r];r++;while (sum < a[l] && r <= m) { // 合并 b 数组的元素sum += b[r];r++;ans++;}l++; // 此时 sum >= a[l]}}cout << ans;return 0;
}