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;
}