题目链接https://www.acwing.com/problem/content/4265/
题目描述:将原序列的连续一段全部增加1或者减少1,求出变成目标序列的最小操作次数
时间复杂度:O(n)
解题思路 (差分)
题目要求同时将一段子数组全部加上1或者减去1, 直觉上就考虑采用差分数组的思想, 将原数组每一项的原始温度减去目标温度, 得到每一项需要改变的数值放在数组a中, 然后求出该数组的差分数组b, 目标是使得差分数组的每一项变成0, 而差分数组的操作存在两种:
- 第一种:对 b 的任一元素加一,另一元素减一(等价于对 a[l,r] 内所有元素加一或减一,若l=r,就是对一个元素操作,此处 r<n,如果r=n,就变成下面一种情况了)
- 第二种:对b的任一元素加一或减一(等价于对 a[l,n] 内所有元素加一或减一,若l=n,就是对最后一个元素操作)
我们每次可以从 b 中同时挑一个正数减 1,和一个负数加 1。也可以只挑一个正数减 1,或一个负数加 1
所以最少次数是 b 中所有正数和和所有负数和的绝对值的较大者。即如果 b 中还有一对正负数就一起操作,否则只能逐个操作
代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N=100010;
int n; //奶牛个数
int a[N]; //气温
int b[N]; //差分数组
void insert(int l,int r,int c)
{
b[l]+=c;
b[r+1]-=c;
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++) cin>>a[i];
for(int i=1;i<=n;i++)
{
int t;
cin>>t; //第i个奶牛的理想气温
a[i]=t-a[i];
}
//构造差分数组
for(int i=1;i<=n;i++)
{
insert(i,i,a[i]);
// cout<<b[i]<<" ";
}
// cout<<endl;
int pos=0,neg=0;
int res;
for(int i=1;i<=n;i++)
{
if(b[i]>0) pos+=b[i];
else neg-=b[i];
}
res=max(pos,neg);
cout<<res<<endl;
return 0;
}