1、用一个队列存数组中所有是m的数的下标,因为所有倒计时肯定是以m开头的
2、然后遍历队列,每次从队头元素的下标开始往下搜,如果搜到倒计时数组,答案 +1
艾弗里有一个由 N 个正整数构成的数组。
数组中的第 i 个整数是 Ai。
如果一个连续的子数组的长度为 m,并且按顺序包含整数 m,m−1,m−2,…,2,1,则称它为 m 倒计数。
例如,[3,2,1] 是 3 倒计数。
请帮助艾弗里计算她的数组中有多少个 K 倒计数。
输入格式
第一行包含整数 T,表示共有 T 组测试数据。
对于每组数据,第一行包含两个整数 N 和 K。
第二行包含 N 个整数,其中第 i 个表示 Ai。
输出格式
每组数据输出一个结果,每个结果占一行。
结果表示为 Case #x: y
,其中 x 为组别编号(从 1 开始),y 为 K 倒计数的数量。
数据范围
1≤T≤100,
2≤K≤N,
1≤Ai≤2×10^5,
2≤N≤2×10^5
输入样例:
3
12 3
1 2 3 7 9 3 2 1 8 3 2 1
4 2
101 100 99 98
9 6
100 7 6 5 4 3 2 1 100
输出样例:
Case #1: 2
Case #2: 0
Case #3: 1
样例解释
在示例 1 中,有两个 3 倒计数,如下:
- 1 2 3 7 9 3 2 1 8 3 2 1
- 1 2 3 7 9 3 2 1 8 3 2 1
在示例 2 中,没有 2 倒计数。
在示例 3 中,有一个 6 倒计数,如下:
- 100 7 6 5 4 3 2 1 100
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 200010;
int n, m;
int a[N], q[N];
int main()
{
int T;
cin >> T; //多组数据
for (int t = 1; t <= T; t ++)
{
cin >> n >> m; //输入n个数,倒计时数
int tt = -1, hh = 0; //数组队列初始化
for (int i = 1; i <= n; i ++)
{
cin >> a[i];
if (a[i] == m) q[++ tt] = i; //因为倒计时出现第一个数肯定是m,所以队列存下所有是m的下标
}
int res = 0; //答案
while (hh <= tt) //队列不空
{
int s = q[hh ++]; //队头弹出,s表示下标
int v = a[s]; //b表示s下标的值
//以为s的位置是倒计时最大的数m,所以下标从s + 1的位置开始搜
//每一个数是上一个数-1
for (int i = s + 1; i <= n; i ++)
{
if (a[i] == v - 1) v = a[i]; //如果当前数是上一个数-1,那就是正确的,将a[i]赋值给v
else break; //如果不是上一个数-1,那就直接break,不会进行了
}
if (v == 1) res ++; //倒计时结束,v的值肯定是1,是1的话,res ++
}
printf("Case #%d: %d\n",t, res); //最后按照样例格式输入答案
}
return 0;
}