Advertisement
MohamedAbdel3al

worms

Feb 14th, 2022
63
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 2.31 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2.  
  3. using namespace std;
  4.  
  5. #define cin(vec) for(auto& i : vec) cin >> i
  6. #define cin_2d(vec, n, m) for(int i = 0; i < n; i++) for(int j = 0; j < m && cin >> vec[i][j]; j++);
  7. #define cout(vec) for(auto& i : vec) cout << i << " "; cout << "\n";
  8. #define cout_2d(vec, n, m) for(int i = 0; i < n; i++, cout << "\n") for(int j = 0; j < m && cout << vec[i][j] << " "; j++);
  9. #define cout_map(mp) for(auto& [f, s] : mp) cout << f << " " << s << "\n";
  10. #define Time cerr << "Time Taken: " << (float)clock() / CLOCKS_PER_SEC << " Secs" << "\n";
  11. #define fixed(n) fixed << setprecision(n)
  12. //#define ceil(n, m) (((n) / (m)) + ((n) % (m) ? 1 : 0))
  13. #define fill(vec, value) memset(vec, value, sizeof(vec));
  14. #define Num_of_Digits(n) ((int)log10(n)+1)
  15. #define all(vec) vec.begin(),vec.end()
  16. #define rall(vec) vec.rbegin(),vec.rend()
  17. #define sz(x) int(x.size())
  18. #define TC int t; cin >> t; while(t--)
  19. #define fi first
  20. #define se second
  21. #define Pair pair < int, int >
  22. #define ll long long
  23. #define ull unsigned long long
  24. #define Mod 1'000'000'007
  25. #define OO 2'000'000'000
  26. #define EPS 1e-9
  27. #define PI acos(-1)
  28. #define imin INT_MIN
  29. #define imax INT_MAX
  30. #define getline(s) getline(cin >> ws , s)
  31. #define pb(x) push_back(x)
  32. #define to_decimal(bin) stoi(bin, nullptr, 2)
  33.  
  34. void Code_Crush(){
  35. ios_base::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
  36. #ifndef ONLINE_JUDGE
  37. freopen("input.txt", "r", stdin), freopen("output.txt", "w", stdout);
  38. #endif
  39. Time
  40. /* MohamedAbdEl3al ♥️🥇♥️ */
  41. }
  42.  
  43. int Binary_Search(vector < int > nums, int test){
  44. int l = 0, r = sz(nums) - 1, ans = 0;
  45. while(l <= r) {
  46. int m = l + (r - l) / 2;
  47. (test <= nums[m] ? r = m - 1, ans = m + 1 : l = m + 1);
  48. }
  49. return ans;
  50. }
  51.  
  52. vector < int > prefix;
  53.  
  54. void prefix_sum(vector < int > nums, int n){
  55. prefix.assign(n + 1, 0);
  56. prefix[0] = nums[0];
  57. for(int i = 1; i < n; i++)
  58. prefix[i] = prefix[i - 1] + nums[i];
  59. }
  60.  
  61. int main (){
  62. Code_Crush();
  63. int n;
  64. cin >> n;
  65. vector < int > nums(n);
  66. for(auto& i : nums) cin >> i;
  67. prefix_sum(nums, n);
  68. int q;
  69. cin >> q;
  70. while(q--){
  71. int test;
  72. cin >> test;
  73. cout << Binary_Search(prefix, test) << '\n';
  74. }
  75. return 0;
  76. }
  77.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement