第一行输"Case #i:"。(由于只有一组样例,只输出”Case #1:”即可)
然后对于每个询问的 k,输出一行包含一个整数,代表数组中满足条件的 k 的大小的区间的数量。
思路:对于每个长度为k的区间只需满足区间最大值减去区间最小值为k-1且没有重复元素就满足条件,所以预处理区间最大值最小值和重复元素的判定,因为k<1000,所以判重复时可以把n^2即10^8的复杂度降为10^7,但其实由于这道题数据太特殊,只有一组数据,这两者差距不是太大。
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#include<string>
#include<map>
#include<set>
using namespace std;
#define LL long long
const int maxn = 10000 + 10;
const int INF = 0x3f3f3f3f;
//freopen("input.txt", "r", stdin);
int a[maxn], b[maxn], dmax[maxn][15], dmin[maxn][15], chongfu[maxn];
void rmq_init(int n) {
for(int i = 0; i < n; i++) {dmax[i][0] = a[i]; dmin[i][0] = a[i];}
for(int j = 1; (1<<j) <= n; j++)
for(int i = 0; i + (1<<j) - 1 < n; i++) {
dmax[i][j] = max(dmax[i][j - 1], dmax[i + (1 << (j - 1))][j - 1]);
dmin[i][j] = min(dmin[i][j - 1], dmin[i + (1 << (j - 1))][j - 1]);
}
}
int qry(int l, int r) {
int k = 0;
while((1 << (k + 1)) <= r - l + 1) k++;
return max(dmax[l][k], dmax[r - (1 << k) + 1][k]) - min(dmin[l][k], dmin[r - (1 << k) + 1][k]);
}
void lisanhua(int n) {
sort(a, a + n);
for(int i = 0; i < n; i++) {
b[i] = lower_bound(a, a + n, b[i]) - a + 1;
}
}
void chongfu_init(int n) { //chongfu[i]数组存的是从i开始到 最长不重复的位置
for(int i = 0; i < n; i++) {
int vis[maxn] = {0};
for(int j = i; j < n && j - i < 1005; j++) {
if(!vis[b[j]]) { chongfu[i] = j; vis[b[j]] = 1;}
else break;
}
}
}
int main() {
int n, m; scanf("%d%d", &n, &m);
for(int i = 0; i < n; i++) {//b数组存的是a的副本
scanf("%d", &a[i]);
b[i] = a[i];
}
rmq_init(n);
lisanhua(n);
chongfu_init(n);
printf("Case #1:\n");
while(m--) {
int k; scanf("%d", &k);
int ans = 0;
for(int i = 0; i + k - 1< n; i++) {
if(qry(i, i + k - 1) == k - 1 && i + k - 1 <= chongfu[i] ) ans++;
}
printf("%d\n", ans);
//for(int i = 0; i < n; i++) for(int j = i; j < n; j++) printf("%d\n", chongfu[i][j]);
}
return 0;
}