码迷,mamicode.com
首页 > 编程语言 > 详细

树状数组---Squared Permutation

时间:2016-05-16 12:24:26      阅读:233      评论:0      收藏:0      [点我收藏+]

标签:

BNUOJ   51636

最近,无聊的过河船同学在玩一种奇怪的名为“小Q的恶作剧”的纸牌游戏。

现在过河船同学手有技术分享张牌,分别写着技术分享,打乱顺序之后排成一行,位置从左往右按照技术分享标号。

接下来小Q同学会给出技术分享个操作,分为以下两种:

1.给定技术分享,交换从左往右数的第技术分享和第技术分享张牌,

2.给定技术分享,对从左往右数的第技术分享张牌,记下位置是这张牌上的数字的牌的数字,询问所有记下的数字加起来的结果。

虽然无聊的过河船同学精通四则运算,但是要完成这么大的计算量还是太辛苦了,希望你能帮他处理这些操作。

 

Input

第一行是一个正整数技术分享,表示测试数据的组数,

对于每组测试数据,

第一行是一个整数技术分享

第二行包含一个技术分享的排列,其中第技术分享个数表示第技术分享张牌上的数字,

第三行是一个整数技术分享,表示操作数,

接下来技术分享行,每行包含三个整数技术分享,其中技术分享表示操作的类型。

Output

对于每组测试数据,依次输出所有查询操作的结果,每个结果一行。

Sample Input

1
3
1 2 3
3
2 1 2
1 1 3
2 2 3

Sample Output

3
5

Hint

对于样例,

第二次操作后牌上的数字从左往右依次是3,2,1,

第三次操作的结果是位置是第2张牌上的数字的牌的数字加上位置是第3张牌上的数字的牌的数字,也就是第2张牌上的数字加上第1张牌上的数字,结果是5。

Source

 
思路:使用树状数组可以方便的求某个连续序列的和;
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
using namespace std;
long long s[100100];
int a[100100],b[100100],flag[100100],t,n,m,Time;
int f[10],tot;

int lowbit(int x)
{
    return x & (-x);
}

void update(int x,long long y)
{
    while(x <= n)
    {
        s[x] += y;
        x += lowbit(x);
    }
}

long long getsum(int x)
{
    long long tmp = 0;
    while(x > 0)
    {
        tmp += s[x];
        x -= lowbit(x);
    }
    return tmp;
}

int main(){
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(int i = 1;i <= n;i++)
        {
            scanf("%d",&a[i]);
            b[a[i]] = i;
            s[i] = 0;
        }
        for(int i = 1;i <= n;i++)update(i,a[a[i]]);
        scanf("%d",&m);
        for(int i = 1;i <= m;i++)
        {
            int op,l,r;
            scanf("%d%d%d",&op,&l,&r);
            if(op == 1)
            {
                Time++;
                tot = 2;
                f[1] = l;
                f[2] = r;
                flag[l] = Time;
                flag[r] = Time;
                if(flag[b[l]] != Time)
                {
                    ++tot;
                    f[tot] = b[l];
                    flag[b[l]] = Time;
                }
                if(flag[b[r]] != Time)
                {
                    ++tot;
                    f[tot] = b[r];
                    flag[b[r]] = Time;
                }
                for(int i = 1;i <= tot;i++)
                update(f[i],-a[a[f[i]]]);
                swap(a[l],a[r]);
                b[a[l]] = l;
                b[a[r]] = r;
                for(int i = 1;i <= tot;i++)
                update(f[i],a[a[f[i]]]);
            }
            else printf("%lld\n",getsum(r)-getsum(l-1));
        }
    }
    return 0;
}

 

树状数组---Squared Permutation

标签:

原文地址:http://www.cnblogs.com/chen9510/p/5497458.html

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