source:7.12模拟赛,原题链接
题目大意:
对于Insert和Delete的操作,显然直接加减即可。对于To的操作,我们容易发现,最优的策略总是光标先往上(下)走,再往左(右)走。考虑到“如果要将其向下(上)移动,而下(上)一行的字符不足a个,那么光标就会落在下(上)一行的最后一个字符后面。”我们只需要维护一个区间最小值即可。这里采用的是单点修改求区间最值的线段树。
附上代码:(要注意long long)的问题
#include<cstdio>
#include<algorithm>
#define maxn 800050
#define INF 2000000000
typedef long long ll;
using namespace std;
struct node
{
int l,r,w;
};
node t[maxn];
char s[10];
ll l[maxn],ans=0,aim;
int n,q,i,now=1,tmp;
void build(int x,int l,int r)
{
t[x].l=l; t[x].r=r;
if (l==r) return;
int mid=(l+r)/2;
build(x*2,l,mid); build(x*2+1,mid+1,r);
t[x].w=min(t[2*x].w,t[2*x+1].w);
}
ll max_(ll x,ll y)
{
if (x>y) return x;else return y;
}
ll min_(ll x,ll y)
{
if (x<y) return x;else return y;
}
void update(int k,int w,int x)
{
if (t[x].l==k&&t[x].r==k)
{
t[x].w=w; return;
}
int mid=(t[x].l+t[x].r)/2;
if (k<=mid) update(k,w,x*2);
else update(k,w,x*2+1);
t[x].w=min(t[x*2].w,t[x*2+1].w);
}
void query(int l,int r,int x)
{
if (t[x].l==l&&t[x].r==r)
{
tmp=min(tmp,t[x].w);
return;
}
int mid=(t[x].l+t[x].r)/2;
if (r<=mid) query(l,r,2*x);
else if (l>mid) query(l,r,2*x+1);
else
{
query(l,mid,2*x);
query(mid+1,r,2*x+1);
}
}
int main()
{
scanf("%d%d",&n,&q);
for (i=1;i<=n;i++) scanf("%lld",&l[i]);
build(1,1,n);
for (i=1;i<=n;i++) update(i,l[i],1);
while (q--)
{
scanf("%s %lld",s,&aim);
if (s[0]=='T')
{
if (aim<now) tmp=INF,query(aim+1,now,1),ans+=now-aim+max(l[aim]-tmp,(ll)0);
if (aim>now) tmp=INF,query(now,aim-1,1),ans+=aim-now+max(l[aim]-tmp,(ll)0);
now=aim;
}
if (s[0]=='I') ans+=aim,update(now,l[now]+aim,1),l[now]+=aim;
if (s[0]=='D') ans+=min(l[now],aim),update(now,max((ll)0,l[now]-aim),1),l[now]-=min(l[now],aim);
}
printf("%lld",ans);
}