função pref
Problem
Dada uma string S
. Encontre a soma dos valores da função de prefixo para todas as posições da string S
.
Entrada
A primeira linha do arquivo de entrada contém a string S
(\(1 <= |S| <= 150 000\) span>) e k
(número de posições especificadas).
Seguido por k
números - posições, cujos valores da função de prefixo devem ser adicionados.
Saída
Saia um único número no arquivo de saída - a soma dos valores da função de prefixo para todas as posições especificadas da string S
.
Exemplos
# |
Entrada |
Saída |
1 |
abacaba 2
3
7
|
4 |