问题描述
小蓝最近获得了一个长度为N 的字符串 S,他对它爱不释手。
小桥为了考验小蓝对字符串的处理能力,决定给他提出一个挑战,她会进行Q次操作:每次操作给定三个整数l,r,k,将S的第l个字符到第r个字符都循环右移k次。
小桥想让小蓝回答她在操作完成后 S是多少?小蓝陷入了困境,于是请你帮帮他!
字符右移表示为按字母表进行移动,例如a右移1次变为b,b右移 2次变为d。特别地,z右移1次变回为 a。
输入格式
第一行输入两个整数 N,Q(1 ≤ N,Q 2 ≤ x10^5)表示字符串 S 的长度和小桥操作次数。
第二行输入一个字符串 S,保证 S 只包含小写字符。
接下来 Q 行每行输入三个整数 l,r,k(1 ≤ l ≤ r≤ N,1 ≤ k ≤ 10^9)表示一次操作。
输出格式
输出一个字符串表示答案。
输入案例
5 3
abcde
1 5 3
1 2 4
2 5 3
输出样例
hlijk
思路想法
其实可以一开始直接使用暴力做法,直接利用for循环对需要改变的区间直接进行改变。
#include <bits/stdc++.h>
using namespace std;
#define int long long
int N,Q;
int k;
char change(char c,int k)
{
return 'a'+(c-'a'+k)%26;
}
signed main()
{
cin>>N>>Q;
string s;
cin>>s;
for(int i=0;i<Q;i++)
{
int l,r,k;
cin>>l>>r>>k;
k%=26;
for(int j=l-1;j<r;j++)
{
s[j]=change(s[j],k);
}
}
cout<<s<<endl;
return 0;
}
由于k的范围过大,暴力方法在后面会超时,于是我们需要优化寻找其他的方法。
差分数组
对于差分,我们可以解决这类问题:给定一个初始数组,然后对其某些区间进行增减操作,最终得到一个最终数组。这个过程中,如果对某个区间进行了增减操作,那么我们希望能够以尽可能高效的方式更新最终数组。
差分的定义:差分数组的第 i 个元素表示的是原数组第 i 个元素与第 i-1 个元素之间的差值。因此,如果原数组为 a,差分数组为 d,那么 d[i] = a[i] - a[i-1]。
更新操作: 对于给定的区间 [l, r],进行增加 k 的操作,相当于在差分数组中,d[l] += k,d[r+1] -= k。这样,当我们进行后续的差分数组前缀和操作时,就可以在 O(1) 的时间内完成区间的增加操作。
差分数组前缀和: 在更新完差分数组后,我们需要还原出最终数组。这时候就可以使用差分数组的前缀和,通过累加差分数组来得到最终数组。即对于差分数组 d,我们可以通过以下方式还原出最终数组 a:
a[0] = d[0]
a[i] = a[i-1] + d[i],对于 i > 0
于是,我们可以这样编写代码框架
vector<int> d(N+1,0); // 初始化差分数组为0
for(int i=0;i<Q;i++)
{
int l,r,k;
cin>>l>>r>>k;
d[l-1]+=k; // 在差分数组中更新区间 (l-1, r)的差分值
d[r]-=k;
}
for(int i=1;i<N;i++)
{
d[i]+=d[i-1]; // 求差分数组的前缀和
}
for(int i=0;i<N;i++)
{
S[i]='a'+(S[i]-'a'+d[i])%26; // 根据差分数组更新字符
}
最终答案
#include <bits/stdc++.h>
using namespace std;
#define int long long
int N,Q;
int k;
signed main()
{
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);//前面三行优化输入输出流的,可以不写
int N,Q;
cin>>N>>Q;
string S;
cin>>S;
vector<int> d(N+1,0);//初始化为0
for(int i=0;i<Q;i++)
{
int l,r,k;
cin>>l>>r>>k;
d[l-1]+=k;
d[r]-=k;
}
for(int i=1;i<N;i++)
{
d[i]+=d[i-1];
}
for(int i=0;i<N;i++)
{
S[i]='a'+(S[i]-'a'+d[i])%26; // 根据差分数组更新字符
}
cout<<S<<endl;
return 0;
}