ACDREAM 02E 小晴天老师系列——我有一个数列!

Problem Description

小晴天:“我有一个数列!”

小晴天:“我还要有很多很多的数列!”

于是小晴天就把这个数列的所有连续子数列写出来。

然后小晴天把每个连续子数列中的最大的数写出来。

那么,有多少个比K大呢?
Input

多组数据,首先是一个正整数t(t<=100),表示数据的组数

对于每组数据,首先是两个整数n(1<=n<=200000),K(0<=K<=10^9).,但所有数据中的n之和不超过1000000.

接下来是n个整数ai

Output
对于每组数据,输出一个整数,表示最大元素大于K的连续子序列的个数。
Sample Input

2
3 2
1 2 3
3 1
1 2 3

Sample Output

3
5

解题思路:dp[i]表示以i为终点,所能找出的最大的符合条件的子序列数量。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#define N 200005
using namespace std;
typedef long long ll;
int num[N];
int dp[N];
int main() {
    int T;
    scanf("%d", &T);
    while (T--) {
        memset(dp, 0, sizeof(dp));
        int n, k;
        scanf("%d %d", &n, &k);
        int cnt = 0;
        for (int i = 0; i < n; i++) {
            scanf("%d", &num[i]);
        }
        if (num[0] > k) dp[0] = 1;
        int sum = dp[0];
        for (int i = 1; i < n; i++) {
            if (num[i] > k) {
                dp[i] = i + 1;
            } else dp[i] = dp[i - 1];
            sum += dp[i];
        }
        printf("%d\n", sum);
    }
    return 0;
}