UVA 12338 - Anti-Rhyme Pairs(접미사 배열 + RMQ)

2121 단어

UVA 12338 - Anti-Rhyme Pairs


제목 링크
제목: 두 문자열의 최대 공통 접두사 길이를 물어볼 때마다 문자열을 지정합니다
사고방식: 문자열을 정렬하면 Height와 rank의 그룹을 구할 수 있고 RMQ를 이용하여 조회하면 된다
코드:
#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <algorithm>
using namespace std;

const int N = 100005;

typedef pair<string, int> pii;

pii str[N];
int save[N];

int t, n, height[N], rank[N];

void init() {
    scanf("%d", &n);
    for (int i = 0; i < n; i++) {
	cin >> str[i].first;
	save[i] = str[i].first.length();
	str[i].second = i;
    }
    sort(str, str + n);
    for (int i = 0; i < n; i++) {
	rank[str[i].second] = i;
	if (i == 0) continue;
	int len = min(str[i - 1].first.length(), str[i].first.length());
	int j;
	for (j = 0; j < len; j++) {
	    if (str[i - 1].first[j] != str[i].first[j])
		break;
	}
	height[i] = j;
    }
}

int best[N * 10][20];

void initRMQ() {
    for (int i = 0; i < n; i++) best[i][0] = height[i];
    for (int j = 1; (1<<j) <= n; j++)
	for (int i = 0; i + (1<<j) - 1 < n; i++)
	    best[i][j] = min(best[i][j - 1], best[i + (1<<(j - 1))][j - 1]);
}

int lcp(int L, int R) {
    L--; R--;
    if (L == R) return save[L];
    L = rank[L]; R = rank[R];
    if (L > R) swap(L, R);
    L++;
    int k = 0;
    while ((1<<(k + 1)) <= R - L + 1) k++;
    return min(best[L][k], best[R - (1<<k) + 1][k]);
}

int main() {
    int cas = 0;
    scanf("%d", &t);
    while (t--) {
	init();
	initRMQ();
	int q, l, r;
	scanf("%d", &q);
	printf("Case %d:
", ++cas); while (q--) { scanf("%d%d", &l, &r); printf("%d
", lcp(l, r)); } } return 0; }

좋은 웹페이지 즐겨찾기