Acwing.241 楼兰图腾（树状数组）

iAkuya

于 2024-03-30 16:30:39 发布

阅读量264

点赞数 5

分类专栏： java算法实录文章标签：算法树状数组

本文链接：https://blog.csdn.net/qq_62235017/article/details/137175273

版权

java算法实录专栏收录该内容

139 篇文章 1 订阅

订阅专栏

文章讲述了如何使用树状数组解决楼兰古城壁画中V和∧图腾的数量计算问题，涉及数据结构和算法优化。

摘要由CSDN通过智能技术生成

题目

在完成了分配任务之后，西部314来到了楼兰古城的西部。

相传很久以前这片土地上(比楼兰古城还早)生活着两个部落，一个部落崇拜尖刀(‘V’)，一个部落祟拜铁锹(‘∧’)，他们分别用V和个的形状来代表各自部落的图腾。

西部314在楼兰古城的下面发现了一幅巨大的壁画，壁画上被标记出了N个点，经测量发现这N个点的水平位置和竖直位置是两两不同的。

西部314认为这幅壁画所包含的信息与这N个点的相对位置有关，因此不妨设坐标分别为(1, y1),(2,y2),… . , (n, yn),其中y1~yn是1到n的一个排列。

西部314打算研究这幅壁画中包含着多少个图腾。

如果三个点(i, yi),(j,yj), (k, yk)满足1 ≤i<j<k ≤n且yi >yj,yj<yk，则称这三个点构成V图腾;如果三个点(i,yi),(j. yj), (k, yk )满足1<i<j<k <n且yi<yj,yj > yk，则称这三个点构成∧图腾;西部314想知道，这n个点中两个部落图腾的数目。

因此，你需要编写一个程序来求出V的个数和∧的个数。

输入格式

第一行一个数n。
第二行是n个数，分别代表y1，y2,… ,yn。

输出格式

两个数，中间用空格隔开，依次为V的个数和∧的个数。

数据范围

对于所有数据,n<200000,且输出答案不会超过int64。

输入样例:

5
1 5 3 2 4

输出样例:

3 4

题解

import java.io.*;
import java.util.Arrays;

/**
 * @author akuya
 * @create 2024-03-30-15:56
 */
public class LoulanMapPreface {
    static int N = 200010;
    static int n;
    static int a[] = new int[N];
    static int tr[] = new int[N];
    static int Greater[] = new int[N];
    static int Lower[] = new int[N];

    public static void main(String[] args) throws IOException {
        BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
        PrintWriter pw=new PrintWriter(new OutputStreamWriter(System.out));
        n=Integer.parseInt(br.readLine());
        String s[]=br.readLine().split(" ");
        for(int i=1;i<=n;i++)
            a[i]=Integer.parseInt(s[i-1]);

        for(int i=1;i<=n;i++){
            int y=a[i];
            Greater[i]=sum(n)-sum(y);
            Lower[i]=sum(y-1);
            add(y,1);
        }

        Arrays.fill(tr,0);
        long res1=0;
        long res2=0;
        for(int i=n;i!=0;i--){
            int y=a[i];
            res1+=Greater[i]*(long)(sum(n)-sum(y));
            res2+=Lower[i]*(long)(sum(y-1));
            add(y,1);
        }

        pw.println(res1+" "+res2);

        br.close();
        pw.flush();
        pw.close();
    }


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

    public static void add(int x,int c){
        for(int i=x;i<=n;i+=lowbit(i)) tr[i]+=c;
    }

    public static int sum(int x){
        int res=0;
        for(int i=x;i!=0;i-=lowbit(i)) res+=tr[i];
        return res;
    }

}