Author : Dipu Kumar Mohanto
CSE, Batch - 6
BRUR.
Problem Statement : 1183 - Computing Fast Average
Source : Light Online Judge
Category : Data Structure
Algorithm : Implicit Treap, Lazy Propagation
Verdict : Accepted
#include "bits/stdc++.h"
using namespace std;
#define FI freopen("in.txt", "r", stdin)
#define FO freopen("out.txt", "w", stdout)
#define FAST ios_base::sync_with_stdio(false), cin.tie(NULL), cout.tie(NULL)
#define FOR(i, n) for (int i = 1; i <= n; i++)
#define For(i, n) for (int i = 0; i < n; i++)
#define ROF(i, n) for (int i = n; i >= 1; i--)
#define Rof(i, n) for (int i = n-1; i >= 0; i--)
#define FORI(i, n) for (auto i : n)
#define ll long long
#define ull unsigned long long
#define vi vector <int>
#define vl vector <ll>
#define pii pair <int, int>
#define pll pair <ll, ll>
#define mk make_pair
#define ff first
#define ss second
#define eb emplace_back
#define em emplace
#define pb push_back
#define ppb pop_back
#define All(a) a.begin(), a.end()
#define memo(a, b) memset(a, b, sizeof a)
#define Sort(a) sort(All(a))
#define ED(a) Sort(a), a.erase(unique(All(a)), a.end())
#define rev(a) reverse(All(a))
//#define sz(a) (int)a.size()
#define max3(a, b, c) max(a, max(b, c))
#define min3(a, b, c) min(a, min(b, c))
#define maxAll(a) *max_element(All(a))
#define minAll(a) *min_element(All(a))
#define allUpper(a) transform(All(a), a.begin(), :: toupper)
#define allLower(a) transform(All(a), a.begin(), :: tolower)
#define endl '\n'
#define nl puts("")
#define ub upper_bound
#define lb lower_bound
#define Exp exp(1.0)
#define PIE 2*acos(0.0)
#define Sin(a) sin(((a)*PIE)/180.0)
#define EPS 1e-9
// int dr[] = {1, -1, 0, 0}; // 4 Direction
// int dc[] = {0, 0, 1, -1};
// int dr[] = {0, 0, 1, -1, 1, 1, -1, -1}; // 8 Direction
// int dc[] = {1, -1, 0, 0, 1, -1, 1, -1};
// int dr[] = {-1, 1, -2, -2, -1, 1, 2, 2}; // knight Moves
// int dc[] = {-2, -2, -1, 1, 2, 2, 1, -1};
#define trace1(x) cerr << #x << ": " << x << endl;
#define trace2(x, y) cerr << #x << ": " << x << " | " << #y << ": " << y << endl;
#define trace3(x, y, z) cerr << #x << ": " << x << " | " << #y << ": " << y << " | " << #z << ": " << z << endl;
#define trace4(a, b, c, d) cerr << #a << ": " << a << " | " << #b << ": " << b << " | " << #c << ": " << c << " | " << #d << ": " << d << endl;
#define trace5(a, b, c, d, e) cerr << #a << ": " << a << " | " << #b << ": " << b << " | " << #c << ": " << c << " | " << #d << ": " << d << " | " << #e << ": " << e << endl;
#define trace6(a, b, c, d, e, f) cerr << #a << ": " << a << " | " << #b << ": " << b << " | " << #c << ": " << c << " | " << #d << ": " << d << " | " << #e << ": " << e << " | " << #f << ": " << f << endl;
inline int setbit(int mask, int pos) { return mask |= (1 << pos); }
inline int resetbit(int mask, int pos) { return mask &= ~(1 << pos); }
inline int togglebit(int mask, int pos) { return mask ^= (1 << pos); }
inline bool checkbit(int mask, int pos) { return (bool)(mask & (1 << pos)); }
#define popcount(mask) __builtin_popcount(mask) // count set bit
#define popcountLL(mask) __builtin_popcountll(mask) // for long long
inline int read() { int a; scanf("%d", &a); return a; }
inline ll readLL() { ll a; scanf("%lld", &a); return a; }
inline double readDD() { double a; scanf("%lf", &a); return a; }
template <typename T> string toString(T num) { stringstream ss; ss << num; return ss.str(); }
int toInt(string s) { int num; istringstream iss(s); iss >> num; return num; }
ll toLLong(string s) { ll num; istringstream iss(s); iss >> num; return num; }
#define inf 1e18
#define mod 1000000007
static const int maxn = 2e5 + 5;
static const int logn = 18;
struct node
{
int val, sum, priority, sze;
int lazy;
node *lft, *rgt;
node(int v = 0)
{
this->val = v;
this->sum = v;
this->priority = rand();
this->sze = 1;
this->lazy = -1;
this->lft = this->rgt = nullptr;
}
};
typedef node* pnode;
inline int sz(pnode t)
{
if (t) return t->sze;
return 0;
}
inline void updateSize(pnode t)
{
if (t) t->sze = sz(t->lft) + 1 + sz(t->rgt);
}
inline void push(pnode t)
{
if (t == nullptr) return;
if (t->lazy != -1)
{
if (t->lft) t->lft->lazy = t->lazy;
if (t->rgt) t->rgt->lazy = t->lazy;
t->val = t->lazy;
t->sum = sz(t->lft) * t->lazy;
t->sum += t->val;
t->sum += sz(t->rgt) * t->lazy;
t->lazy = -1;
}
}
inline void combine(pnode t)
{
if (t == nullptr) return;
push(t->lft);
push(t->rgt);
t->sum = t->val;
if (t->lft) t->sum += t->lft->sum;
if (t->rgt) t->sum += t->rgt->sum;
}
inline void Split(pnode t, pnode &l, pnode &r, int pos, int add = 0)
{
if (t == nullptr) return void(l = r = nullptr);
push(t);
int curr = sz(t->lft) + add;
if (curr <= pos) Split(t->rgt, t->rgt, r, pos, curr+1), l = t;
else Split(t->lft, l, t->lft, pos, add), r = t;
updateSize(t);
combine(t);
}
inline void Merge(pnode &t, pnode l, pnode r)
{
push(l);
push(r);
if (l == nullptr || r == nullptr) t = l ? l : r;
else if (l->priority < r->priority) Merge(r->lft, l, r->lft), t = r;
else Merge(l->rgt, l->rgt, r), t = l;
updateSize(t);
combine(t);
}
inline void Insert(pnode &t, pnode newnode, int pos)
{
pnode L, R;
Split(t, L, R, pos-1);
Merge(t, L, newnode);
Merge(t, t, R);
}
inline pnode Update(pnode t, int l, int r, int val)
{
if (t == nullptr) return t;
pnode L, mid, R;
Split(t, L, mid, l-1);
Split(mid, t, R, r-l);
t->lazy = val;
Merge(mid, L, t);
Merge(t, mid, R);
return t;
}
inline int query(pnode t, int l, int r)
{
if (!t) return 0;
pnode L, mid, R;
Split(t, L, mid, l-1);
Split(mid, t, R, r-l);
int ans = t->sum;
Merge(mid, L, t);
Merge(t, mid, R);
return ans;
}
inline void display(pnode t)
{
if (t == nullptr) return;
display(t->lft);
trace1(t->val);
display(t->rgt);
}
inline void clean(pnode t)
{
if (t == nullptr) return;
clean(t->lft);
clean(t->rgt);
delete t;
}
int main()
{
int tc = read();
FOR(tcase, tc)
{
int n = read();
int q = read();
pnode treap = nullptr;
For(i, n)
{
pnode t;
Merge(t, treap, new node(0));
treap = t;
}
//display(treap);
printf("Case %d:\n", tcase);
while (q--)
{
int type = read();
if (type == 1) // change
{
int l = read();
int r = read();
int val = read();
if (l > r) swap(l, r);
Update(treap, l, r, val);
}
else
{
int l = read();
int r = read();
if (l > r) swap(l, r);
int sum = query(treap, l, r);
int len = r - l + 1;
int gcd = __gcd(sum, len);
sum /= gcd;
len /= gcd;
printf("%d", sum);
if (len > 1) printf("/%d", len);
puts("");
}
}
clean(treap);
}
}
No comments:
Post a Comment
Note: Only a member of this blog may post a comment.