Advertisement
Sergey101

Q4

Jan 30th, 2025
28
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.89 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. typedef int_fast64_t lli;
  4.  
  5. int main()
  6. {
  7.     ios_base::sync_with_stdio(false);
  8.     cin.tie(nullptr);
  9.     cout.tie(nullptr);
  10.  
  11.     lli n, m;
  12.     cin >> n >> m;
  13.     vector<lli> a(n);
  14.     vector<lli> b(m);
  15.     for(lli i = 0; i < n; i++)
  16.     {
  17.         cin >> a[i];
  18.     }
  19.     for(lli i = 0; i < m; i++)
  20.     {
  21.         cin >> b[i];
  22.         b[i]--;
  23.     }
  24.  
  25.     lli ans = 0;
  26.     lli acc = 0;
  27.     for(lli i = m-1; i >= 0; i--)
  28.     {
  29.         if(a[b[i]] > 0 && acc > 0)
  30.         {
  31.             a[b[i]] = -1;
  32.             ans += 2;
  33.             acc--;
  34.         }
  35.         else
  36.         {
  37.             acc++;
  38.         }
  39.     }
  40.  
  41.     lli c = 0;
  42.     for(lli i = 0; i < n; i++)
  43.     {
  44.         c += a[i]+1;
  45.     }
  46.    
  47.     if(c > acc)
  48.     {
  49.         cout << -1 << '\n';
  50.     }
  51.     else
  52.     {
  53.         cout << ans + c << '\n';
  54.     }
  55.     return 0;
  56. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement