uvalive 4329 Ping pong---树状数组

来源:互联网 发布:劈照片的软件 编辑:程序博客网 时间:2024/06/07 01:26

Problem :n(3<=n<=20000) 个乒乓球爱好者,每个人有一个技能值,他们的每场比赛位两个选手一个裁判,裁判必住在两个选手之间,并且技能值也在两个选手之间(比一个小比另一个大),一共能组织多少种比赛。

Way :如果第i个人当裁判,假设a1到a(i-1)有lelt[i]个比ai小,那么就有(i-1)-left[i]个比a[i]大,同理a(i+1)到a[n],有right[i]个比a[i]小,就有(n-i)-right[i]个比a[i]大,则第i个人当裁判有:left[i]*(n-i-right[i])+(i-left[i]-1)*right[i]种比赛,只需

求出left[i]和right[i]即可,x[i] = 1表示已有a[i],详见代码:

code:

////  main.cpp//  Calmeii//  树状数组//  Created by Calmeii on 14-5-7.//  Copyright (c) 2014年 apple. All rights reserved.//#include <stdio.h>#include <string.h>#include <iostream>#include <algorithm>using namespace std;#define MAXN 20010  //数字最大个数#define lowbit(v) ((v)&(-v))// a存放数值,r,l分别存放右边和左边比i小的个数,x存放前缀和,mx最大的那个数int a[MAXN], r[MAXN], l[MAXN], n, x[100010], mx;int sum(int id){    int ans = 0;    while (id > 0)    {        ans += x[id];        id -= lowbit(id);    }    return ans;}void add(int id, int v){    while (id <= mx)    {        x[id] += v;        id += lowbit(id);    }}int main(){    int t;    scanf("%d", &t);    while (t--)    {        int n;        mx = 0;        scanf("%d", &n);        for (int i = 1; i <= n; i++){            scanf("%d", &a[i]);            mx = max(mx, a[i]);        }        memset(x, 0, sizeof(x));    //x的初始化        for (int i = 1; i <= n; i++){            add(a[i],1);    //前缀和加            l[i] = sum(a[i]-1);   //i左边的比a[i]小的个数        }        memset(x, 0, sizeof(x));    //x的初始化        for (int i = n; i >= 1; i--){            add(a[i], 1);            r[i] = sum(a[i]-1);     //i右边的比a[i]小的个数        }        long long ans = 0;        for (int i = 1; i <= n; i++){            ans += l[i]*(n-i-r[i]) + (i-l[i]-1)*r[i];        }        printf("%lld\n", ans);    }        return 0;}/*263 2 8 4 9 2*/



0 0
原创粉丝点击