Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include<bits/stdc++.h>
- using namespace std;
- int main() {
- int a,b,ans=0,cnt=0;
- cin>>a>>b;
- string s,s1;
- cin>>s;
- char ch;
- map<char,int>mp;
- while(b--) {
- cin>>ch;
- mp[ch]=1;
- }
- vector<int>v;
- for(int i=0; i<a; i++) {
- if(mp[s[i]]==1) {
- v.push_back(1);
- } else
- v.push_back(0);
- }
- for(int j=0; j<a; j++) {
- if(v[j]==1) {
- cnt++;
- }
- if(v[j]==0) {
- ans+=((cnt*(cnt+1))/2);
- cnt=0;
- }
- }
- cout<<ans<<endl;
- }
Advertisement
Add Comment
Please, Sign In to add comment