C. Perform the Combo
先把操作排序,然后二分找大于等于当前 i 的操作数有多少,对应的字符加上这个数就行。
#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字符的个数
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;
}