C. Perform the Combo
time limit per test2 seconds
memory limit per test256 megabytes
inputstandard input
outputstandard output
You want to perform the combo on your opponent in one popular fighting game. The combo is the string s consisting of n lowercase Latin letters. To perform the combo, you have to press all buttons in the order they appear in s. I.e. if s=“abca” then you have to press ‘a’, then ‘b’, ‘c’ and ‘a’ again.
You know that you will spend m wrong tries to perform the combo and during the i-th try you will make a mistake right after pi-th button (1≤pi<n) (i.e. you will press first pi buttons right and start performing the combo from the beginning). It is guaranteed that during the m+1-th try you press all buttons right and finally perform the combo.
I.e. if s=“abca”, m=2 and p=[1,3] then the sequence of pressed buttons will be ‘a’ (here you’re making a mistake and start performing the combo from the beginning), ‘a’, ‘b’, ‘c’, (here you’re making a mistake and start performing the combo from the beginning), ‘a’ (note that at this point you will not perform the combo because of the mistake), ‘b’, ‘c’, ‘a’.
Your task is to calculate for each button (letter) the number of times you’ll press it.
You have to answer t independent test cases.
Input
The first line of the input contains one integer t (1≤t≤104) — the number of test cases.
Then t test cases follow.
The first line of each test case contains two integers n and m (2≤n≤2⋅105, 1≤m≤2⋅105) — the length of s and the number of tries correspondingly.
The second line of each test case contains the string s consisting of n lowercase Latin letters.
The third line of each test case contains m integers p1,p2,…,pm (1≤pi<n) — the number of characters pressed right during the i-th try.
It is guaranteed that the sum of n and the sum of m both does not exceed 2⋅105 (∑n≤2⋅105, ∑m≤2⋅105).
It is guaranteed that the answer for each letter does not exceed 2⋅109.
Output
For each test case, print the answer — 26 integers: the number of times you press the button ‘a’, the number of times you press the button ‘b’, …, the number of times you press the button ‘z’.
Example
inputCopy
3
4 2
abca
1 3
10 5
codeforces
2 8 3 2 9
26 10
qwertyuioplkjhgfdsazxcvbnm
20 10 1 2 3 5 10 5 9 4
outputCopy
4 2 2 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
0 0 9 4 5 3 0 0 0 0 0 0 0 0 9 0 0 3 1 0 0 0 0 0 0 0
2 1 1 2 9 2 2 2 5 2 2 2 1 1 5 4 11 8 2 7 5 1 10 1 5 2
Note
The first test case is described in the problem statement. Wrong tries are “a”, “abc” and the final try is “abca”. The number of times you press ‘a’ is 4, ‘b’ is 2 and ‘c’ is 2.
In the second test case, there are five wrong tries: “co”, “codeforc”, “cod”, “co”, “codeforce” and the final try is “codeforces”. The number of times you press ‘c’ is 9, ‘d’ is 4, ‘e’ is 5, ‘f’ is 3, ‘o’ is 9, ‘r’ is 3 and ‘s’ is 1.
首先将所有前缀,出现次数存到二维数组中,之后模拟
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<stdio.h>
#include<iostream>
#include<map>
using namespace std;
int T;
int p[300000];
int dp[300000][30];
int vis[30];
string s;
int main()
{
std::ios::sync_with_stdio(false);
cin.tie();
cin >> T;
int n,m;
while (T--)
{
cin >> n>>m;
cin >> s;
for (int i = 0; i < 30; i++)
dp[0][i] = 0;
dp[0][s[0] - 'a']++;
for (int i = 1; i < s.size(); i++)
{
for (int j = 0; j < 26; j++)dp[i][j] = dp[i - 1][j];
dp[i][s[i] - 'a']++;
}
memset(vis, 0, sizeof(vis));
int t = -1;
for (int i = 0; i < m; i++)
{
cin >> p[i];
if (t == -1 && p[i] >= n)
t = i;
}
p[m] = n;
if (t == -1)
t = m+1;
for (int i = 0; i < t; i++)
{
for (int j = 0; j < 26; j++)
vis[j] += dp[p[i]-1][j];
}
for (int i = 0; i < 26; i++)
{
cout << vis[i];
if (i != 25)
cout << " ";
}
cout << endl;
}
return 0;
}