[Hiho一下 第230周] Smallest Substring（贪心,RMQ)

发表于 | 更新于 | 分类于 ， | 评论数： | 阅读次数：

题目链接：http://hihocoder.com/contest/hiho230/problem/1
题意：找一个字符串中长度为k的字典序最小的子串。

考虑贪心，每次选中整个串的最小字符，那么这个字符一定属于答案，这个字符右边的最小字符也一定属于答案。当处理完这个字符后继时，看看是否取够，如果没取够那么再处理这个字符的前驱，直到取够为止。

于是需要用ST表解决一下静态区间最值查询的问题。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#include <bits/stdc++.h>
using namespace std;

const int maxn = 100100;
int n, k;
char a[maxn];
int dp[maxn][20];
vector<int> ret;

int gao(int x, int y) {
  if(a[x] < a[y]) return x;
  if(a[x] == a[y]) return x < y ? x : y;
  return y;
}

void st() {
  for(int i = 1; i <= n; i++) dp[i][0] = i;
  int k = int(log(n+1.0)/log(2.0));
  for(int j = 1; j <= k; j++) {
    for(int i = 1; i + (1 << j) - 1 <= n; i++) {
      dp[i][j] = gao(dp[i][j-1], dp[i+(1<<(j-1))][j-1]);
    }
  }
}

int query(int l, int r) {
  int k = int(log(r-l+1.0)/log(2.0));
  return gao(dp[l][k], dp[r-(1<<k)+1][k]);
}

void dfs(int l, int r) {
  if(ret.size() == k || l > r) return;
  int id = query(l, r);
  ret.emplace_back(id);
  dfs(id+1, r);
  dfs(l, id-1);
}

signed main() {
  // freopen("in", "r", stdin);
  while(~scanf("%d%s",&k,a+1)) {
    n = strlen(a+1);
    st();
    ret.clear();
    dfs(1, n);
    sort(ret.begin(), ret.end());
    for(int i = 0; i < ret.size(); i++) {
      printf("%c", a[ret[i]]);
    }
    printf("\n");
  }
}