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