Codeforces Round #368 (Div. 2) E. Garlands 二维树状数组

最新推荐文章于 2022-08-21 01:05:54 发布

acblacktea

最新推荐文章于 2022-08-21 01:05:54 发布

阅读量341

点赞数

分类专栏：树状数组

本文链接：https://blog.csdn.net/acblacktea/article/details/52317913

版权

树状数组专栏收录该内容

4 篇文章 0 订阅

订阅专栏

枚举每条链把每条链上的点的价值都加到树状数组去，然后枚举询问把每个包括这条链的一部分的询问的答案添加这部分答案，用树状数组查询。。。。
时间复杂度 q*k*log(n)*log(m) 可怕明显超时啊cf机子跑得太快了居然都过了

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<vector>
#include<utility>
#define LL long long
#define maxn 2005
using namespace std;
LL va[maxn][maxn];

vector<pair<pair<LL, LL>, LL> > block[maxn]; // x,y,va

vector<int>up[maxn];

struct node{
     int x1, y1, x2, y2, flag;
     LL ans;
}que[maxn * 1005];

int n,m,l = 0, k;
char s[20];

void update(int X, int Y, LL tmp)
{
     for(int i = X; i <=n; i+=(i&(-i)))
         for(int j = Y; j<=m ; j+=(j&(-j)))
             va[i][j] += tmp;
}

LL query(int X, int Y)
{
   LL ans = 0;
   for(int i = X; i; i-=(i&(-i)))
       for(int j = Y; j; j-=(j&(-j)))
           ans += va[i][j];
   return ans;
}

int main()
{
    scanf("%d %d %d", &n, &m, &k);
    for(int i = 1; i<= k; i++)
    {
        int len; scanf("%d", &len);
        for(int j = 0; j < len; j++)
        {
            int X1,Y1; LL V;
            scanf("%d %d %I64d", &X1, &Y1, &V);
            block[i].push_back(make_pair(make_pair(X1, Y1), V));
        }
    }

    int q;
    scanf("%d", &q);
    for(int i = 0; i < q; i++)
    {
        scanf("%s", s);
        if(s[0] == 'A')
        {
           scanf("%d %d %d %d",&que[l].x1, &que[l].y1, &que[l].x2, &que[l].y2);
           que[l].flag = i;
           l++;
        }
        else{
            int a;
            scanf("%d", &a);
            up[a].push_back(i);
        }
    }

    for(int i = 1;i <= k; i++)
    {
        up[i].push_back(10000000);
        for(int j = 0; j < block[i].size(); j++)
        {
            update(block[i][j].first.first, block[i][j].first.second, block[i][j].second);
        }

        int pre = 0, sym = 1;
        for(int j = 0; j < l; j++)
        {
            while(que[j].flag > up[i][pre])
            {
                sym ^= 1;
                pre++;
            }

            if(sym)
            {
                int X1 = que[j].x1, Y1 = que[j].y1, X2 = que[j].x2, Y2 = que[j].y2;
                que[j].ans += query(X2, Y2) - query(X2, Y1 - 1) - query(X1 - 1, Y2) + query(X1 - 1, Y1 - 1);
            }
        }

        for(int j = 0; j < block[i].size(); j++)
        {
            update(block[i][j].first.first, block[i][j].first.second, -1 * block[i][j].second);
        }
    }

    for(int i = 0; i < l; i++) printf("%I64d\n", que[i].ans);

    return 0;
}