Advertisement
GerONSo

Untitled

Nov 4th, 2019
116
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 5.95 KB | None | 0 0
  1. /*
  2.  
  3. ∧_∧
  4. ( ・ω・。)つ━☆・*。
  5. ⊂  ノ    ・゜
  6. しーJ   Accepted
  7.  
  8. */
  9.  
  10.  
  11.  
  12. // #pragma GCC optimize("O3")
  13. // #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
  14.  
  15. #include <bits/stdc++.h>
  16. #include <ext/pb_ds/assoc_container.hpp>
  17. #include <ext/pb_ds/tree_policy.hpp>
  18.  
  19. #define ll long long
  20. #define all(x) begin(x), end(x)
  21. #define x first
  22. #define y second
  23. #define int long long
  24.  
  25. using namespace std;
  26. using namespace __gnu_pbds;
  27.  
  28. typedef long double ld;
  29. template<typename T>
  30. using kawaii_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
  31.  
  32. const ld PI = atan2(0, -1);
  33.  
  34. void seriy() {
  35. ios::sync_with_stdio(0);
  36. cin.tie(0);
  37. cout.tie(0);
  38. cout << fixed << setprecision(14);
  39. #ifdef _offline
  40. freopen("input.txt", "r", stdin);
  41. freopen("output.txt", "w", stdout);
  42. #else
  43. freopen("river.in", "r", stdin);
  44. freopen("river.out", "w", stdout);
  45. #endif
  46. }
  47.  
  48. const int MAXN = 1e6 + 10;
  49. const int MAXM = 600;
  50. const int INF = 1e18;
  51. const int BASE = 47;
  52. const int MOD = 1e9 + 7;
  53. const int MAXLOG = 21;
  54.  
  55. mt19937 rr(time(0));
  56.  
  57. struct treap {
  58. treap *left = nullptr;
  59. treap *right = nullptr;
  60. int sum, prior, sz, val;
  61.  
  62. treap(int sum) {
  63. this->sum = sum;
  64. this->val = sum;
  65. this->prior = rr();
  66. this->sz = 1;
  67. }
  68. };
  69.  
  70.  
  71. int get_size(treap *&root) {
  72. return ((root == nullptr) ? 0 : root->sz);
  73. }
  74.  
  75. int get_val(treap *&root) {
  76. return ((root == nullptr) ? 0 : root->sum);
  77. }
  78.  
  79. int get_val2(treap *&root) {
  80. return ((root == nullptr) ? 0 : root->val);
  81. }
  82.  
  83. void print(treap *a) {
  84. if(a->left != nullptr) print(a->left);
  85. cerr << get_val2(a) << " " << get_val2(a->left) << " " << get_val2(a->right) << " " << get_size(a) << '\n';
  86. if(a->right != nullptr) print(a->right);
  87. }
  88.  
  89. void update(treap *&root) {
  90. if(root == nullptr) return;
  91. root->sz = get_size(root->left) + get_size(root->right) + 1;
  92. root->sum = get_val(root->left) + get_val(root->right) + root->val;
  93. }
  94.  
  95. pair<treap*, treap*> split(treap *&root, int key) {
  96. if(root == nullptr) {
  97. return {nullptr, nullptr};
  98. }
  99. int root_key = get_size(root->left);
  100. // cerr << root->sz << " " << root_key << '\n';
  101. if(key > root_key) {
  102. pair<treap*, treap*> splitted = split(root->right, key - get_size(root->left) - 1);
  103. root->right = splitted.x;
  104. update(root);
  105. return {root, splitted.y};
  106. }
  107. else {
  108. pair<treap*, treap*> splitted = split(root->left, key);
  109. root->left = splitted.y;
  110. update(root);
  111. return {splitted.x, root};
  112. }
  113. }
  114.  
  115. treap* merge(treap *&left, treap *&right) {
  116. if(left == nullptr) {
  117. return right;
  118. }
  119. if(right == nullptr) {
  120. return left;
  121. }
  122. if(left->prior > right->prior) {
  123. left->right = merge(left->right, right);
  124. update(left);
  125. return left;
  126. }
  127. else {
  128. right->left = merge(left, right->left);
  129. update(right);
  130. return right;
  131. }
  132. }
  133.  
  134. void insert(treap *&root, int pos, int val) {
  135. pair<treap*, treap*> splitted = split(root, pos);
  136. treap *nw = new treap(val * val);
  137. splitted.x = merge(splitted.x, nw);
  138. update(splitted.x);
  139. // cerr << get_val(left) << '\n';
  140. splitted.x = merge(splitted.x, splitted.y);
  141. update(splitted.x);
  142. // cerr << get_val(left) << '\n';
  143. root = splitted.x;
  144. }
  145.  
  146. void erase(treap *&root, int pos) {
  147. if(pos < get_size(root->left)) {
  148. erase(root->left, pos);
  149. }
  150. else if(pos > get_size(root->left)) {
  151. erase(root->right, pos - get_size(root->left) - 1);
  152. }
  153. else {
  154. root = merge(root->left, root->right);
  155. }
  156. update(root);
  157. }
  158.  
  159. int get(treap *&root, int pos) {
  160. if(pos < get_size(root->left)) {
  161. return get(root->left, pos);
  162. }
  163. else if(pos > get_size(root->left)) {
  164. return get(root->right, pos - get_size(root->left) - 1);
  165. }
  166. else {
  167. return get_val2(root);
  168. }
  169. }
  170.  
  171. signed main() {
  172. seriy();
  173. int n, p;
  174. cin >> n >> p;
  175. vector<int> a(n);
  176. treap *root = nullptr;
  177. // print(root);
  178. int cnt = n;
  179. pair<treap*, treap*> splitted = split(root, 0);
  180. for(int i = 0; i < n; i++) {
  181. cin >> a[i];
  182. insert(root, i, a[i]);
  183. // cerr << root->sz << '\n';
  184. }
  185. // print(root);
  186. cout << root->sum << '\n';
  187. int q;
  188. cin >> q;
  189. while(q--) {
  190. int t, pos;
  191. cin >> t >> pos;
  192. pos--;
  193. if(t == 2) {
  194. int w = sqrt(get(root, pos));
  195. erase(root, pos);
  196. insert(root, pos, (w + 1) / 2);
  197. insert(root, pos, w / 2);
  198. cnt++;
  199. }
  200. else {
  201. if(pos == 0) {
  202. int a = sqrt(get(root, 0));
  203. int b = sqrt(get(root, 1));
  204. erase(root, 0);
  205. erase(root, 0);
  206. int nv = (a + b);
  207. // cerr << nv << '\n';
  208. insert(root, 0, nv);
  209. }
  210. else if(pos == cnt - 1) {
  211. int a = sqrt(get(root, cnt - 1));
  212. int b = sqrt(get(root, cnt - 2));
  213. erase(root, cnt - 2);
  214. erase(root, cnt - 2);
  215. int nv = a + b;
  216. insert(root, cnt - 2, nv);
  217. }
  218. else {
  219. int a = sqrt(get(root, pos - 1));
  220. int b = sqrt(get(root, pos + 1));
  221. int w = sqrt(get(root, pos));
  222. erase(root, pos - 1);
  223. erase(root, pos - 1);
  224. erase(root, pos - 1);
  225. int nv1 = b + (w + 1) / 2;
  226. int nv2 = a + w / 2;
  227. insert(root, pos - 1, nv1);
  228. insert(root, pos - 1, nv2);
  229. }
  230. cnt--;
  231. }
  232. cout << root->sum << '\n';
  233. }
  234. return 0;
  235. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement