Module: تابع پیشوند، تابع Z


Problem

3 /10


تابع pref

Problem

یک رشته S داده می شود. مجموع مقادیر تابع پیشوند را برای همه موقعیت های داده شده رشته S پیدا کنید. 

ورودی
خط اول فایل ورودی حاوی رشته S است (\(1 <= |S| <= 150 000\)) و (تعداد موقعیت های مشخص شده).
به دنبال اعداد k - موقعیت هایی که مقادیر تابع پیشوند آن باید اضافه شود.

خروجی
خروجی یک عدد واحد در فایل خروجی - مجموع مقادیر تابع پیشوند برای تمام موقعیت های مشخص شده رشته S.
 

 

نمونه‌ها
<سر> <بدن>
# ورودی خروجی
1
abacaba 2
3
7
4