1. 程式人生 > >牛客練習賽3 E-絕對半徑2051(尺取/離散化+二分)

牛客練習賽3 E-絕對半徑2051(尺取/離散化+二分)

題目描述

????是一名狙擊手,憑藉肉眼視覺可以做到精確命中絕對半徑2051公尺的一切目標。 作為一名優秀的狙擊手,????不僅經常保養槍支,也經常保養彈藥。 ????有?枚子彈,第?枚的型號為??,????打算扔掉其中最多?枚。 大多數優秀的狙擊手都有藝術癖好,????希望扔掉一部分子彈後,最 長的連續相同子彈序列的長度儘量長。

輸入描述:

第一行,兩個整數?,?。
第二行,?個正整數??。

輸出描述:

一行,一個整數,最長的連續相同子彈序列的長度。

示例1

輸入

複製

8 1
1 1 1 2 2 3 2 2

輸出

複製

4

備註:

對於10%的資料,? ≤ 10。
對於30%的資料,? ≤ 1000。
對於60%的資料,?? ≤ 30。
對於100%的資料,0 ≤ ? ≤ ? ≤ 105,1 ≤ ?? ≤ 109。

法一:尺取法

我們把尺取的起點和終點作為我們得到的區間,從前往後尺取並用map來維護區間所有數字出現的次數,尺取左端對應的數字一定是需要連續的數字,然後我們記錄一個最大的答案即可,當右端點移動到不滿足條件時就移動左端點並更新map

AC程式碼:

#include<bits/stdc++.h>
#define bug printf("*********\n");
#define mem0(a) memset(a, 0, sizeof(a));
#define mem1(a) memset(a, -1, sizeof(a));
#define finf(a, n) fill(a, a+n, INF);
#define pb(G, b) G.push_back(b);
#define lowbit(i) i&(-i)
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
typedef long long LL;
typedef unsigned long long uLL;
typedef pair<LL, pair<int, LL> > LLppar;
typedef pair<int, int> par;
typedef pair<LL, int> LLpar;
const LL mod = 1e9+7;
const LL INF = 1e9+7;
const uLL base1 = 131;
const uLL base2 = 233;
const int MAXN = 300010;
const int MAXM = 300010;
const double pi = acos(-1);
 
int n, k;
int a[100010];
map<int, int> mmp;
 
int main() {
    while(~scanf("%d%d", &n, &k)) {
        mmp.clear();
        for(int i = 0; i < n; i ++) {
            scanf("%d", &a[i]);
        }
        int l = 0, r = -1, ans = 0;
        while(l < n) {
            while(r < n && r-l+1-mmp[a[l]] <= k) {
                mmp[a[++ r]] ++;
            }
            ans = max(ans, mmp[a[l]]);
            while(l < n && a[l] == a[l+1]) { //更新左端點到另一個值
                mmp[a[l]] --;
                l ++;
            }
            mmp[a[l]] --;
            l ++;
        }
        printf("%d\n", ans);
    }
    return 0;
}

法二:二分+離散化

由於數字很大,然而數字的個數很小,所以我們離散化一下數字,然後用vector來裝每一種數字出現位置,並列舉每種數字在每一個出現的位置作為終點位置,最後二分起點位置即可。

離散化有兩種方法,map的好像慢一點

for(int i = 0; i < n; i ++) {
    scanf("%d", &a[i]);
    b[i] = a[i];
}
sort(b, b+n);
int cnt = unique(b, b+n)-b;
for(int i = 0; i < n; i ++) {
    a[i] = lower_bound(b, b+cnt, a[i])-b;
    v[a[i]].push_back(i);
}
for(int i = 0; i < n; i ++) {
    scanf("%d", &a[i]);
    if(!mmp[a[i]]) {
        mmp[a[i]] = ++ cnt;
    }
    v[mmp[a[i]]].push_back(i);
}

AC程式碼:

#include<bits/stdc++.h>
#define bug printf("*********\n");
#define mem0(a) memset(a, 0, sizeof(a));
#define mem1(a) memset(a, -1, sizeof(a));
#define finf(a, n) fill(a, a+n, INF);
#define pb(G, b) G.push_back(b);
#define lowbit(i) i&(-i)
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
typedef long long LL;
typedef unsigned long long uLL;
typedef pair<LL, pair<int, LL> > LLppar;
typedef pair<int, int> par;
typedef pair<LL, int> LLpar;
const LL mod = 1e9+7;
const LL INF = 1e9+7;
const uLL base1 = 131;
const uLL base2 = 233;
const int MAXN = 300010;
const int MAXM = 300010;
const double pi = acos(-1);

int n, k;
int a[100010];
map<int, int> mmp;
vector<int> v[100010];

int main() {
    //while(~scanf("%d%d", &n, &k)) {
        scanf("%d%d", &n, &k);
        mmp.clear();
        int cnt = 0;
        for(int i = 0; i < n; i ++) {
            scanf("%d", &a[i]);
            if(!mmp[a[i]]) {
                mmp[a[i]] = ++ cnt;
            }
            v[mmp[a[i]]].push_back(i);
        }
        int ans = 0;
        for(int i = 1; i <= cnt; i ++) {
            for(int j = 0; j < v[i].size(); j ++) {
                int l = 0, r = j, s = 0;
                while(l <= r) {
                    int mid = (l+r)/2;
                    if(v[i][j]-v[i][mid]+1-(j-mid+1) <= k) {
                        s = mid;
                        r = mid-1;
                    }else l = mid+1;
                }
                ans = max(ans,j-s+1);
            }
        }
        printf("%d\n", ans);
    //}
    return 0;
}