1. 程式人生 > >【題解】CQOI2015任務查詢系統

【題解】CQOI2015任務查詢系統

nod node bool ger -- oid 上進 while div

  主席樹,操作上面基本上是一樣的。每一個時間節點一棵樹,一個樹上的每個節點代表一個優先級的節點。把開始和結束時間點離散,在每一棵樹上進行修改。註意因為一個時間節點可能會有多個修改,但我們要保證都在同一棵樹上,所以我采取了讓每個節點額外存儲所屬於的樹的一個信息。當更新到一個節點的時候,如果屬於建立好的新樹,那麽就修改;不屬於,則重新建鏈。

  但這題有一個地方需要註意,就是所求的個數有可能小於一個葉子結點中存儲的任務個數,這時我們就要返回k * num。

代碼(有的地方感覺自己寫的有些啰嗦,大家看看就好):

// luogu-judger-enable-o2
#include <bits/stdc++.h>
using
namespace std; #define int long long #define maxn 1500000 int n, m, timer, tot, cnt, sum, pre = 1, root[maxn], Ti[maxn]; map <int, int> Map; bool vis[maxn]; struct node { int t, num; }upd[maxn * 2]; struct tree { int lson, rson, cal, root, size, rank; }T[maxn * 5]; struct project { int
s, e, p; }P[maxn]; bool cmp(project a, project b) { return a.p < b.p; } bool cmp2(node a, node b) { return a.t < b.t; } int read() { int x = 0, k = 1; char c; c = getchar(); while(c < 0 || c > 9) { if(c == -) k = -1; c = getchar(); } while(c >= 0 && c <=
9) x = x * 10 + c - 0, c = getchar(); return x * k; } void update(int &now, int bf, int x, int num, int l, int r, int root) { if(root != T[now].root) now = ++ sum, T[now] = T[bf], T[now].root = root; if(num > 0) T[now].size += 1; else T[now].size -= 1; if(l == r) { T[now].cal += num; T[now].rank = num; return; } int mid = (l + r) >> 1; if(x <= mid) update(T[now].lson, T[bf].lson, x, num, l, mid, root); else update(T[now].rson, T[bf].rson, x, num, mid + 1, r, root); T[now].cal = T[T[now].lson].cal + T[T[now].rson].cal; } int query(int now, int L, int R, int k) { int mid = (L + R) >> 1; int size = T[T[now].lson].size; if(L == R) return k * T[now].rank; if(k >= T[now].size) return T[now].cal; if(k <= size) return query(T[now].lson, L, mid, k); else return query(T[now].lson, L, mid, size) + query(T[now].rson, mid + 1, R, k - size); } signed main() { m = read(), n = read(); for(int i = 1; i <= m; i ++) { P[i].s = read(), P[i].e = read(), P[i].p = read(); upd[++ cnt] = (node) {P[i].s, P[i].p}; upd[++ cnt] = (node) {P[i].e + 1, -P[i].p}; } sort(P + 1, P + 1 + m, cmp); for(int i = 1; i <= m; i ++) if(P[i].p != P[i - 1].p) Map[P[i].p] = ++ tot; sort(upd + 1, upd + 1 + cnt, cmp2); for(int i = 1; i <= cnt; i ++) { if(upd[i].t != upd[i - 1].t) { int now = ++ timer; Ti[upd[i].t] = now; root[timer] = ++ sum; int t = upd[i].t - 1; while(!Ti[t]) Ti[t] = now - 1, t --; } int t = Ti[upd[i].t]; update(root[t], root[t - 1], Map[abs(upd[i].num)], upd[i].num, 1, tot, t); vis[t] = true; } for(int i = 1; i <= n; i ++) { int x = read(), a = read(), b = read(), c = read(); int ans, k = 1 + (a * pre + b) % c; printf("%lld\n", ans = query(root[Ti[x]], 1, tot, k)); pre = ans; } return 0; }

【題解】CQOI2015任務查詢系統