library

Some useful algorithms for competitive programming

  1. 1
  2. 2
  3. 3
  4. 4
  5. 5
  6. 6
  7. 7
  8. 8
  9. 9
  10. 10
  11. 11
  12. 12
  13. 13
  14. 14
  15. 15
  16. 16
  17. 17
  18. 18
  19. 19
  20. 20
  21. 21
  22. 22
  23. 23
  24. 24
  25. 25
  26. 26
  27. 27
  28. 28
  29. 29
  30. 30
  31. 31
  32. 32
  33. 33
  34. 34
  35. 35
  36. 36
  37. 37
  38. 38
  39. 39
  40. 40
struct y_node {
	int val;
	y_node* c[2];
	y_node() { val = 0; c[0] = c[1] = 0; }
	y_node* get_c(int i) { return (!c[i] ? c[i] = new y_node : c[i]); }
	void update(int y, int v, int l = -1, int r = MX) {
		if (l == r) val += v;
		else {
			int m = (l + r) >> 1;
			y <= m ? get_c(0)->update(y, v, l, m) : get_c(1)->update(y, v, m + 1, r);
			val = (c[0] ? c[0]->val : 0) + (c[1] ? c[1]->val : 0);
		}
	}
	int query(int yl, int yr, int l = -1, int r = MX) {
		if (yl > r || yr < l) return 0;
		if (yl <= l && yr >= r) return val;
		int m = (l + r) >> 1;
		return (c[0] ? c[0]->query(yl, yr, l, m) : 0) + (c[1] ? c[1]->query(yl, yr, m + 1, r) : 0);
	}
};

struct x_node {
	y_node seg;
	x_node* c[2];
	x_node() { c[0] = c[1] = 0;}
	x_node* get_c(int i) { return (!c[i] ? c[i] = new x_node : c[i]); }
	void update(int x, int y, int v, int l = -1, int r = MX) {
		if (l == r) seg.update(y, v);
		else {
			int m = (l + r) >> 1;
			x <= m ? get_c(0)->update(x, y, v, l, m) : get_c(1)->update(x, y, v, m + 1, r);
			seg.update(y, (c[0] ? c[0]->seg.query(y, y) : 0) + (c[1] ? c[1]->seg.query(y, y) : 0));
		}
	}
	int query(int xl, int xr, int yl, int yr, int l = -1, int r = MX) {
		if (xl > r || xr < l) return 0;
		if (xl <= l && xr >= r) return seg.query(yl, yr);
		int m = (l + r) >> 1;
		return (c[0] ? c[0]->query(xl, xr, yl, yr, l, m) : 0) + (c[1] ? c[1]->query(xl, xr, yl, yr, m + 1, r) : 0);
	}
} root;