码迷,mamicode.com
首页 > 其他好文 > 详细

【题解】CQOI2015任务查询系统

时间:2018-03-08 02:33:59      阅读:184      评论:0      收藏:0      [点我收藏+]

标签:有一个   define   系统   ==   enable   地方   signed   str   for   

  主席树,操作上面基本上是一样的。每一个时间节点一棵树,一个树上的每个节点代表一个优先级的节点。把开始和结束时间点离散,在每一棵树上进行修改。注意因为一个时间节点可能会有多个修改,但我们要保证都在同一棵树上,所以我采取了让每个节点额外存储所属于的树的一个信息。当更新到一个节点的时候,如果属于建立好的新树,那么就修改;不属于,则重新建链。

  但这题有一个地方需要注意,就是所求的个数有可能小于一个叶子结点中存储的任务个数,这时我们就要返回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任务查询系统

标签:有一个   define   系统   ==   enable   地方   signed   str   for   

原文地址:https://www.cnblogs.com/twilight-sx/p/8525725.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!