fork download
  1. #include <iostream>
  2. #include <cmath>
  3. using namespace std;
  4.  
  5. int main() {
  6. int n , k , a[10001] , s[10001] ;
  7. cin >> n >> k ;
  8.  
  9. long long sum = 0 , sum1 = 0;
  10. for(int i = 0 ; i< n ; i++)
  11. {
  12. cin >> a[i] ;
  13. }
  14. int c =0 ;
  15. while(k--)
  16. {
  17. cin >> s[c];
  18. abs(sum -=s[c]);
  19. c++;
  20.  
  21. }
  22.  
  23. for(int i = a[sum] ; i<n ; i++)
  24. {
  25. if(i >= 0 && i<n-1)
  26. {
  27. sum1+=a[i];
  28. }
  29. else{
  30. for(int j = a[sum] ; j > 0 ; j--)
  31. {
  32.  
  33. sum1+=a[j];
  34. }
  35.  
  36.  
  37.  
  38.  
  39. }
  40.  
  41.  
  42.  
  43. }
  44.  
  45.  
  46. cout << sum1 << "\n";
  47.  
  48.  
  49.  
  50. return 0;
  51. }
Success #stdin #stdout 0s 15240KB
stdin
4 1
2 3 1 2
3
stdout
6