C_Yet_Another_Broken_Keyboard.cpp

#include <bits/stdc++.h>

#define ll long long
#define ld long int
#define vi vector<int>
#define pii pair<int, int>
#define pb push_back

using namespace std;

// 8""""8
// 8    " eeeee eeeee eeee
// 8e     8  88 8   8 8
// 88     8   8 8e  8 8eee
// 88   e 8   8 88  8 88
// 88eee8 8eee8 88ee8 88ee

// --------MATHS OPERATIONS---------

const int N = 2e5 + 1;
const int MOD = 1e9 + 7;
ll fact[N], inv[N], invfact[N];
void factInverse()
{
    fact[0] = inv[1] = fact[1] = invfact[0] = invfact[1] = 1;
    for (long long i = 2; i < N; i++)
    {
        fact[i] = (fact[i - 1] * i) % MOD;
        inv[i] = MOD - (inv[MOD % i] * (MOD / i) % MOD);
        invfact[i] = (inv[i] * invfact[i - 1]) % MOD;
    }
}

int add(int a, int b)
{
    if ((a += b) >= MOD)
        a -= MOD;
    else if (a < 0)
        a += MOD;
    return a;
}

ll mul(int x, int y)
{
    return (1LL * x * y) % MOD;
}

ll nCr(int n, int r)
{
    if (r > n)
        return 0;
    return mul(mul(fact[n], invfact[r]), invfact[n - r]);
}

long long pow(long long base, long long n, long long m = MOD)
{

    long long ans = 1ll;
    while (n != 0)
    {
        if (n % 2)
        {
            ans = (ans * base) % m;
            n--;
        }
        else
        {
            base = (base * base) % m;
            n /= 2;
        }
    }
    return ans;
}

// --------SOLVE---------

void solve()
{
    ld n, k;
    cin >> n >> k;
    string input;
    cin >> input;
    char check[k];
    for (int i = 0; i < k; i++)
    {
        cin >> check[i];
    }
    ll o = 0;
    ll result = 0;
    for (ld j = 0; j < n; j++)
    {
        bool isPresent = false;
        for (int i = 0; i < k; i++)
        {
            if (input[j] == check[i])
            {
                isPresent = true;
                break;
            }
        }
        if (isPresent)
        {
            o++;
        }
        else
        {
            if (o)
            {
                result += (o * (o + 1)) / 2;
            }
            o = 0;
        }
    }

    if (o)
    {
        result += (o * (o + 1)) / 2;
    }
    cout << result;
}

// --------XXXXXXXXX---------

int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);

    int t = 1;
    //  cin >> t;
    while (t--)
    {
        solve();
    }

    return 0;
}