C. Perform the Combo

先把操作排序,然后二分找大于等于当前 i 的操作数有多少,对应的字符加上这个数就行。

AC代码:

#include <bits/stdc++.h>
using namespace std;
const int N=2e5+10;
char s[N];
int n,m,T,ans[30],p[N];
int main()
{
    ios::sync_with_stdio(false);
    cin>>T;
    while(T--)
    {
        cin>>n>>m>>s+1;
        m++;
        for(int i=1;i<=m-1;i++)
            cin>>p[i];
        p[m]=n;
        sort(p+1,p+m+1);
        memset(ans,0,sizeof(ans));
        for(int i=1;i<=n;i++)
        {
            int k=lower_bound(p+1,p+m+1,i)-p;
            ans[s[i]-'a']+=m+1-k;
        }
        for(int i=0;i<26;i++)
            i==25?printf("%d\n",ans[i]):printf("%d ",ans[i]);
    }
    return 0;
}

前缀和计数做法会超时:

#include <bits/stdc++.h>
using namespace std;
const int N=2e5+10;
char s[N];
int n,m,T,x,sum[N][150],ans[30],t[150];//sum[i][j]表示到第i个时,字符j的个数
int main()
{
    ios::sync_with_stdio(false);;
    cin>>T;
    while(T--)
    {
        cin>>n>>m;
        memset(sum,0,sizeof(sum));
        memset(ans,0,sizeof(ans));
        memset(t,0,sizeof(t));
        for(int i=1;i<=n;i++)
        {
            cin>>s[i];
            t[s[i]]++;
            for(int j=0;j<26;j++)
                sum[i][j+'a']=t[j+'a'];
        }
        for(int i=1;i<=m+1;i++)
        {
            if(i<m+1)cin>>x;
            else x=n;
            for(int j=0;j<26;j++)
                ans[j]+=sum[x][j+'a'];
        }
        for(int i=0;i<26;i++)
            i==25?printf("%d\n",ans[i]):printf("%d ",ans[i]);
    }
    return 0;
}

超时的原因应该是数组空间比较大,memset比较耗时。
以后用memset要考虑数组空间,多组数据的时候尽量减小数组空间(要改进算法)。