TYVJ1432 楼兰图腾

P1432 [Tyvj Jan]楼兰图腾

时间: 1000ms / 空间: 131072KiB / Java类名: Main

描述

在完成了分配任务之后，西部314来到了楼兰古城的西部。相传很久以前这片土地上(比楼兰古城还早)生活着两个部落，一个部落崇拜尖刀(‘V’)，一个部落崇拜铁锹(‘∧’)，他们分别用V和∧的形状来代表各自部落的图腾。
西部314在楼兰古城的下面发现了一幅巨大的壁画，壁画上被标记出了N个点，经测量发现这N个点的水平位置和竖直位置是两两不同的。西部314认为这幅壁 画所包含的信息与这N个点的相对位置有关，因此不妨设坐标分别为(1,y1),(2,y2),…,(n,yn),其中y1~yn是1到n的一个排列。
西部314打算研究这幅壁画中包含着多少个图腾，其中V图腾的定义如下（注意：图腾的形式只和这三个纵坐标的相对大小排列顺序有关）1<=i<j<k<=n且yi>yj,yj<yk;

而崇拜∧的部落的图腾被定义为1<=i<j<k<=n且yi<yj,yj>yk;

西部314想知道，这n个点中两个部落图腾的数目。因此，你需要编写一个程序来求出V的个数和∧的个数。

输入格式

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

输出格式

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

测试样例1

输入

5
1 5 3 2 4

输出

3 4

备注

样例解释

数据范围
10%的数据 n<=600
40%的数据 n<=5000
100%的数据 n<=200000，答案不超过int64西部314admin  tyvj首届月赛第四道

【题解】

我们发现无论是枚举左边一个点还是右边一个点，都很难统计，于是我们枚举中间一个点，

找左边比他大/小的，右边比他大/小的，乘起来，加合即可

这个题数据有问题

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <queue>
#include <vector>
#define max(a, b) ((a) > (b) ? (a) : (b))
#define min(a, b) ((a) < (b) ? (a) : (b))
#define lowbit(a) ((a) & (-1 * (a)))

inline void read(long long &x)
{
    x = 0;char ch = getchar(), c = ch;
    while(ch < '0' || ch > '9')c = ch, ch = getchar();
    while(ch <= '9' && ch >= '0')x = x * 10 + ch - '0', ch = getchar();
    if(c == '-')x = -x;
}

const long long MAXN = 1000000 + 10;

long long n, num[MAXN], bst[MAXN], tmp1[MAXN], tmp2[MAXN], ma;

void modify(long long p, long long k)
{
    for(;p <= ma;p += lowbit(p))bst[p] += k; 
}

long long ask(long long p)
{
    long long sum = 0;
    for(;p;p -= lowbit(p))sum += bst[p];
    return sum;
}

int main()
{
    read(n);
    for(register long long i = 1;i <= n;++ i) read(num[i]), ma = max(ma, num[i]);
    register long long ans = 0;
    for(register long long i = 1;i <= n;++ i)
    {
        if(num[i] < 0 || num[i] > n)continue;
        tmp1[i] = ask(n - num[i]);
        modify(n - num[i] + 1, 1);
    }
    memset(bst, 0, sizeof(bst));
    for(register long long i = n;i >= 1;-- i)
    {
        if(num[i] < 0 || num[i] > n)continue;
        tmp2[i] = ask(n - num[i]);
        modify(n - num[i] + 1, 1);
    }
    for(register long long i = 1;i <= n;++ i)ans += tmp1[i] * tmp2[i];
    printf("%lld ", ans);
    ans = 0;
    memset(bst, 0, sizeof(bst));
    for(register long long i = 1;i <= n;++ i)
    {
        if(num[i] < 0 || num[i] > n)continue;
        tmp1[i] = ask(num[i] - 1);
        modify(num[i], 1);
    }
    memset(bst, 0, sizeof(bst));
    for(register long long i = n;i >= 1;-- i)
    {
        if(num[i] < 0 || num[i] > n)continue;
        tmp2[i] = ask(num[i] - 1); 
        modify(num[i], 1);
    }
    for(register long long i = 1;i <= n;++ i)ans += tmp1[i] * tmp2[i];
    printf("%lld", ans);
    return 0;
}