zoukankan      html  css  js  c++  java
  • poj-3928(树状数组)

    题目链接:传送门

    题意:n个乒乓球运动员要互相练习,都去一个运动员那里比赛,举办训练的运动员不能水平最高或最低。

    现在给出n个运动员的水平,求出最终有多少种组合。

    思路:先对运动员进行离散化,然后进行比较,第一位和最后一位运动员的不能举行训练,所以跳过,

    对剩下的运动员进行查找,有两种情况

    (1)这个运动员的前面的人的水平比当前的人低的人数有pr,后面水平比前面高的人有lt个,ans+=pr*lt。

    (2)这个运动员的前面的人的水平比当前的人高的人数有pr,后面水平比前面低的人有lt个,ans+=pr*lt。

    所以求出最终的人数。

    注意:

    (1)最终结果很多,ans要用long long类型。

    (2)要先对第一名运动员进行update操作。

    #include<iostream>
    #include<cstdio>
    #include<cstring>
    #include<algorithm>
    using namespace std;
    const int maxn = 100100*4;
    typedef long long LL;
    int c[maxn],n,b[maxn];
    struct Node{
        int id,x;
    }a[maxn];
    bool cmp(Node a,Node b)
    {
        return a.x<b.x;
    }
    int lowbit(int x)
    {
        return x&(-x);
    }
    void update(int x)
    {
        while(x<maxn)
        {
            c[x]++;
            x+=lowbit(x);
        }
    }
    int query(int x)
    {
        int sum=0;
        while(x>0)
        {
            sum+=c[x];
            x-=lowbit(x);
        }
        return sum;
    }
    int main(void)
    {
        int T,i,j,n;
        scanf("%d",&T);
        while(T--)
        {
            memset(b,0,sizeof(b));
            memset(c,0,sizeof(c));
            scanf("%d",&n);
            for(i=1;i<=n;i++) scanf("%d",&a[i].x),a[i].id=i;
            sort(a+1,a+n+1,cmp);
            for(i=1;i<=n;i++) b[a[i].id]=i;
            LL ans=0,pr,lt;
            update(b[1]);
            for(i=2;i<n;i++)
            {
                update(b[i]);
                int tmp=query(b[i]);
                pr=tmp-1;
                lt=n-i-b[i]+tmp;
                ans+=pr*lt;
                pr=i-tmp;
                lt=b[i]-tmp;
                ans+=pr*lt;
            }
            printf("%lld
    ",ans);
        }
        return 0;
    }
    View Code
  • 相关阅读:
    luogu P4852 yyf hates choukapai 单调队列优化dp
    luogu P2605 [ZJOI2010]基站选址 线段树优化dp
    7.9 NOI模拟赛 数列 交互 高精 字符串
    AGC 043 C
    7.9 NOI模拟赛 A.图 构造 dfs树 二分图
    7.9 NOI模拟赛 C.走路 背包 dp 特异性
    2020 高考记
    7.6 NOI模拟赛 灯 根号分治
    python-08-元组
    python-07-列表
  • 原文地址:https://www.cnblogs.com/2018zxy/p/10187649.html
Copyright © 2011-2022 走看看