ZOJ 3790 Consecutive Blocks (离散化 + 暴力)

题目链接

虽然是一道暴力的题目，但是思路不好想。刚开始还超时，剪枝了以后1200ms，不知道为什么还是这么慢。

题意：给你n个点，每个点有一种颜色ci，给你至多k次删除操作，每次删除一个点，问最多k次操作后连在一起的点颜色相同的最大长度。

思路：由于ci 最大为10^9, 所以需要首先离散化。然后暴力。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <map>
#include <vector>
#include <algorithm>
#define LL long long
using namespace std;
const int maxn = 100000 + 10;
vector<int>v[maxn];
int c[maxn], ans, n, k;

void cal(int x)
{
    int i, sz = v[x].size();
    int sum, p;
    if(sz <= ans)
    return;
    for(i = 0; i < sz; i++) //暴力枚举开始的连续的一串都为x的数字。
    {
        sum = 0;
        for(p = i; p < sz-1; p++)
        {
            if(sum+v[x][p+1]-v[x][p] - 1>k)
            break;
            sum += v[x][p+1] - v[x][p] - 1;
        }
        sum = v[x][p] - v[x][i] + 1 - sum;
        ans = max(ans, sum);
        if(p == sz-1)
        break;
    }
}
int main()
{
    int i, a, cnt;
    while(~scanf("%d%d", &n, &k))
    {
        ans = 1;
        map<int, int>mp;
        cnt = 1;
        for(i = 0; i < n; i++)
        {
            scanf("%d", &a);
            if(mp[a] == 0) mp[a] = cnt++; //离散化 cnt.
            c[i] = mp[a];
            v[mp[a]].clear();
        }
        for(i = 0; i < n; i++) v[c[i]].push_back(i);
        for(i = 1; i < cnt; i++)
        cal(i);
        printf("%d\n", ans);
    }
    return 0;
}