原题链接:
1987. 粉刷栅栏 - AcWing题库高质量的算法题库https://www.acwing.com/problem/content/1989/
思路:
哈希+差分。本题虽然n比较小,但a[i]的规模很大,所以想到用map储存下标,奶牛移动无非也就是给某个区间都加1,这就想到差分了,但注意一点,差分map中的点是离散的,要还原成原数组需要求前缀和,但由于值域规模很大,所以直接求前缀和肯定超时,所以需要注意差分数组的性质,如下图所示:
所以会发现只需处理出现在差分数组里的元素即可,一个区间点的数量就是两个端点下标的差值。注意用map存的时候处理差分时不要mp[r+1]--,因为遍历时需要用到的端点是r不是r+1,因为我们默认以l为区间开头,所以会有1的偏移量,而且一个区间内元素的个数与这个区间的开闭无关
Code:
#include <iostream>
#include <cstring>
#include <algorithm>
#include <map>
using namespace std;
map<int,int> mp;
void insert(int l,int r)
{
mp[l]++;
mp[r]--;
}
int main()
{
int n;
cin>>n;
int poi=0,las;
while (n -- )
{
int x;
char c;
cin>>x>>c;
if(c=='L')
{
las=poi;
poi-=x;
insert(poi,las);
}
else
{
las=poi;
poi+=x;
insert(las,poi);
}
}
int last=0,ans=0,sum=0;
for(auto& [x,v]:mp)
{
if(sum>=2) ans+=x-last;
last=x;
sum+=v;
}
cout<<ans<<endl;
return 0;
}
作者:机械之忍
链接:https://www.acwing.com/activity/content/code/content/2288002/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。