HDU 5247 - 找連續數 ( 思路 + RMQ )

題意

一個長度爲n ( n <= 10^4 )的無序數組, 找是否存在k區間使得區間內元素排序後是連續的, 求有多少個這樣的k區間( k <= 1000 )

思路

RMQ算法預處理區間的思想實現O(n)查詢
記錄這個區間的最大最小值, 如果 最大值-最小值+1 == j-i+1 則 ans[j-i+1]++

剪枝:
1. 如果 最大值 - 最小值>1000, break
2. 如果 某個數在這個區間內出現次數(用map記錄)大於一次, break

一開始a[]數組開小了返的tle, 本以爲如果是這樣應該是re的, 記一下返t還有可能是因爲越界

AC代碼

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <map>
#include <cmath>
using namespace std;
//typedef unsigned long long ll;
const int maxn = 1e4+5;
const int mmaxn = 1e3+5;
const int INF = 0x3f3f3f3f;
int a[maxn];
int ans[mmaxn];
map<int, int> mp;
int n;

void solve(){
    int minmrk, maxmrk, i, j;
    for(i = 1; i <= n; i++){
        mp.clear();
        minmrk = a[i];
        maxmrk = a[i];
        mp[a[i]]++;
        ans[1]++;
        for(j = i+1; j <= n; j++){
            minmrk = min(minmrk, a[j]);
            maxmrk = max(maxmrk, a[j]);
            mp[a[j]]++;
            if( mp[a[j]] > 1 || maxmrk - minmrk > 1000 ) break;
            if( maxmrk - minmrk == j - i )
                ans[j-i+1]++;
        }
    }
}

void check(){
    puts("------");
    for( int i = 1; i <= n; i++ )
        printf("%d : %d \n", i, ans[i]);
    puts("------");
}

int main()
{
    int m, k;
    int kase = 0;
    while( ~scanf("%d%d",&n,&m) ){
        memset(ans, 0, sizeof ans);
        for( int i = 1; i <= n; i++ )
            scanf("%d",&a[i]);
        solve();
        //check();
        printf("Case #%d:\n", ++kase);
        while(m--){
            scanf("%d", &k);
            printf("%d\n", ans[k]);
        }
    }
    return 0;
}