Advertisement
saurav_kalsoor

Charges AC

Jun 1st, 2021
56
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.03 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. #define ll long long
  3. #define MOD 1000000007
  4. #define endl '\n'
  5.  
  6. using namespace std;
  7.  
  8. void solve(){
  9. int n, k;
  10. cin>>n>>k;
  11. string s;
  12. cin>>s;
  13.  
  14. vector<int> q(k);
  15. for(int i=0;i<k;i++){
  16. cin>>q[i];
  17. }
  18.  
  19.  
  20. ll res = 0;
  21. for(int i=0; i < n-1 ; i++){
  22. if(s[i] == s[i+1])
  23. res += 2;
  24. else
  25. res++;
  26. }
  27.  
  28. for(int j=0; j < k; j++){
  29. int i = q[j] - 1;
  30. if(s[i] == '1')
  31. s[i] = '0';
  32. else
  33. s[i] = '1';
  34.  
  35. if(i > 0){
  36. if(s[i-1] == s[i])
  37. res++;
  38. else
  39. res--;
  40. }
  41.  
  42. if(i < n-1){
  43. if(s[i+1] == s[i])
  44. res++;
  45. else
  46. res--;
  47. }
  48. cout<<res<<endl;
  49. }
  50.  
  51. }
  52.  
  53. signed main() {
  54. ios_base::sync_with_stdio(false);
  55. cin.tie(NULL);
  56.  
  57. int t;
  58. cin>>t;
  59. while(t--)
  60. solve();
  61.  
  62. }
  63.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement