2015百度之星初賽(1)·找連續數·RMQ

題目大意:

小度熊拿到了一個無序的數組,對於這個數組,小度熊想知道是否能找到一個k 的區間,裏面的 k 個數字排完序後是連續的。 

現在小度熊增加題目難度,他不想知道是否有這樣的 k 的區間,而是想知道有幾個這樣的 k 的區間。

解題思路:

如果這樣一個區間滿足題目要求,那麼這個區間滿足兩個條件:

1、最大值減最小值 = K - 1;

2、區間內無重複數字;

對於一點我們很容易想到是區間最值的問題,而第二點的處理由於數據量的原因無法n^2暴力,因此我們可以採取某些技巧:

設數組F[i],表示第i位最長可以延伸到哪個位置保證區間內無重複數字,那樣的話,由於題目說了查詢區間大小不會超過1000,因此最大複雜度爲O(1000n),貌似沒降下多少,但是還是能過得,卡的不嚴所以網上很多都是BRUTE FORCE算法做的。

AC代碼:

#include <map>
#include <set>
#include <stack>
#include <cmath>
#include <queue>
#include <bitset>
#include <string>
#include <vector>
#include <cstdio>
#include <cctype>
#include <fstream>
#include <cstdlib>
#include <sstream>
#include <cstring>
#include <iostream>
#include <algorithm>
#pragma comment(linker, "/STACK:1024000000,1024000000")

using namespace std;
//#define   maxd          1010
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
#define   mc(x, y)     memcpy(x, y, sizeof(x))
#define   ms(x,y)      memset(x,y,sizeof(x))
#define   rep(i,n)      for(int i=0;i<(n);i++)
#define   repf(i,a,b)   for(int i=(a);i<=(b);i++)
#define   PI           pair<int,int>
//#define   mapp            make_pair
#define   FI            first
#define   SE            second
#define   IT            iterator
#define   PB            push_back
#define   Times         10
typedef   long long     ll;
typedef   unsigned long long ull;
typedef   long double   ld;
typedef   pair<int,int > pce;
//#define N 100
const double eps = 1e-10;
const double  pi = acos(-1.0);
const  ll    mod = 1e9+7;
const  int   inf = 0x3f3f3f3f;
//const  ll    INF = (ll)1e18+300;
const int   maxd = 50500 + 10;

int n, Q;
int ac[maxd];
int max_[maxd][20];
int min_[maxd][20];
map<int, int > mp;
int vis[maxd];

void init() {
    for (int i = 1; i <= n; i++) {
        max_[i][0] = min_[i][0] = ac[i];
    }
    for (int i = 1; i <= n; i++) {
        mp.clear();
        for (int j = i; j <= n && j - i <= 1001; j++) {
            if (mp[ac[j]]) {
                break;
            }
            vis[i] = j;
            mp[ac[j]] = 1;
        }
    }
}
void RMQ() {
    for (int j = 1; j <= log2(n + 1); j++) {
        for (int i = 1; i <= n; i++) {
            if(i + (1<<j) - 1 <= n) {
                max_[i][j] = max(max_[i][j-1], max_[i + (1<<(j-1))][j-1]);
                min_[i][j] = min(min_[i][j-1], min_[i + (1<<(j-1))][j-1]);
            }
        }
    }
}

int query(int L, int R) {
    int k = (int )(log(R - L + 1) / log(2.0));
    int max_ans = max(max_[L][k], max_[R - (1<<k) + 1][k]);
    int min_ans = min(min_[L][k], min_[R - (1<<k) + 1][k]);
    return max_ans - min_ans;
}
int main() {
    int kase = 0;
    while(cin >> n >> Q) {
        ms(vis, 0 );
        kase ++;
        for (int i = 1; i <= n; i++) {
            cin >> ac[i];
        }
        cout << "Case #" << kase << ":" << endl;
        init();
        RMQ();
        while(Q --) {
            int k;
            cin >> k;
            int ans = 0;
            for (int i = 1; i <= n - k + 1; i++) {
                int cnt = query(i, i + k - 1);
                if(cnt == k - 1 && i + k <= vis[i] + 1) {
                    ans ++;
                }
            }
            cout << ans << endl;
        }
    }
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章