蓝桥杯 算法提高 天天向上

问题描述

　　A同学的学习成绩十分不稳定，于是老师对他说：“只要你连续4天成绩有进步，那我就奖励给你一朵小红花。”可是这对于A同学太困难了。于是，老师对他放宽了要求：“只要你有4天成绩是递增的，我就奖励你一朵小红花。”即只要对于第i、j、k、l四天，满足i<j<k<l并且对于成绩wi<wj<wk<wl，那么就可以得到一朵小红花的奖励。现让你求出，A同学可以得到多少朵小红花。

输入格式

　　第一行一个整数n，表示总共有n天。第二行n个数，表示每天的成绩wi。

输出格式

　　一个数，表示总共可以得到多少朵小红花。

样例输入

6
1 3 2 3 4 5

样例输出

6

#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long LL;
typedef pair<int,int>pii;
const int mod = 1e9 + 7 , INF = 0x3f3f3f3f , N = 2e3 + 10;

int w[N];
int large[N]; 
int lower[N];
// large[i] : 在 i 后面并且比 i 大的元素的个数 
// lower[i] : 在 i 前面并且比 i 小的元素的个数
// 递增四元组下标为 a,b,c,d
// f[b][c] = lower(a) + large(d)
int main()
{   
    int n;
    cin >> n;
    for (int i = 1 ; i <= n ; i ++)
        cin >> w[i];
    for (int i = 2 ; i <= n - 1 ; i ++)
    {
        for (int j = 1 ; j < i ; j ++)
            if (w[i] > w[j])
                lower[i] ++;
        for (int j = i + 1 ; j <= n ; j ++)
            if (w[i] < w[j]) 
                large[i] ++;
    }       
    LL res = 0;
    for (int i = 2 ; i <= n - 2 ; i ++)
    for (int j = i + 1 ; j <= n - 1 ; j ++)
        if (w[j] > w[i])
            res += lower[i] * large[j];
    cout << res << endl;
    return 0;
}