题意:
小晴天:“我有一个数列!”
小晴天:“我还要有很多很多的数列!”
于是小晴天就把这个数列的所有连续子数列写出来。
然后小晴天把每个连续子数列中的最大的数写出来。
那么,有多少个比K大呢?
Input
多组数据,首先是一个正整数t(t<=100),表示数据的组数
对于每组数据,首先是两个整数n(1<=n<=200000),K(0<=K<=10^9).,但所有数据中的n之和不超过1000000.
接下来是n个整数a[i](1<=a[i]<=10^9)
Output
对于每组数据,输出一个整数,表示最大元素大于K的连续子序列的个数。
Sample Input
2 3 2 1 2 3 3 1 1 2 3
Sample Output
3 5
Hint
对于样例一,共有6个连续子序列{1}{2}{3}{1,2}{2,3}{1,2,3}(注意{1,3}不满足题意,因为不连续)
其中最大元素大于2的共有3个{3}{2,3}{1,2,3}
对于样例二,大于1的连续子序列共有5个,{2}{3}{1,2}{2,3}{1,2,3}
题解:找到所有比k大的数字的位置存到pos[i]里,然后从头考虑每个数字,如果比k小,从它出发的第一个连续序列必须是到它右面的最近的比k大的数字,然后对于每个数字只要把它右边的解都计算出来就不会出现重复,所以直接(n - pos[i] + 1) * (pos[i] - pos[i - 1])可以直接计算所有解。
#include <stdio.h>
const int N = 200005;
int n, k, s[N], pos[N];
int main() {
int t;
scanf("%d", &t);
while (t--) {
scanf("%d%d", &n, &k);
int cnt = 1;
for (int i = 1; i <= n; i++) {
scanf("%d", &s[i]);
if (s[i] > k)
pos[cnt++] = i;
}
long long res = 0;
for (int i = 1; i < cnt; i++)
res += (n - pos[i] + 1) * (pos[i] - pos[i - 1]);
printf("%lld\n", res);
}
return 0;
}