題意

給你一個數字序列S，再給一個數字序列pattern，S和pattern中的數字都是1到s（s<=25)。每個序列里的數字都有個排名，也就是第幾小，現在我們要用pattern來匹配S。在本題中匹配是指每個數字的排名都一樣，即同是第k小，最后輸出匹配的所有位置。

思路

KMP好題，對KMP的理解又透徹了一點點~ 我們考慮兩個字符串A，B 在此題中，A[1],A[2],…,A[k]與B[1],B[2],…,B[k]匹配條件： 若A[1],A[2],…,A[k-1]與B[1],B[2],…,B[k-1]匹配，則加上A[k]與B[k]仍然匹配的條件是： 必須在與k無關的時間復雜度內完成該操作 然而，由于A[1],A[2],…,A[k-1]與B[1],B[2],…,B[k-1]已經匹配，可以簡化比較操作 考慮到字符集很小（不超過25），我們可以定義以下幾個變量： Occur[p],Low[i],High[i]可能不存在，若存在則取最小的符合條件的x 則A[1],A[2],…,A[k-1]與B[1],B[2],…,B[k-1]匹配，加上A[k]與B[k]仍然匹配的條件可簡化為： 接下來就可以用O(SK)時間內預處理求出Occur[p],Low[i],High[i]，然后套用kmp算法，總時間復雜度為O(N+SK)

代碼

? [cpp] #include <iostream> #include <cstdio> #include <cmath> #include <algorithm> #include <string> #include <cstring> #include <vector> #include <set> #include <stack> #include <queue> #define MID(x,y) ((x+y)/2) #define MEM(a,b) memset(a,b,sizeof(a)) #define REP(i, begin, end) for (int i = begin; i <= end; i ++) using namespace std; typedef long long LL; int n, k, s; int S[100005], pattern[25005]; int low[25005], high[25005], ranks_pos[30], next[100005]; vector <int> ans; void init(){ MEM(ranks_pos, -1); MEM(low, -1); MEM(high, -1); for (int i = 0; i < k; ++ i){ int j = pattern[i] - 1; while(j){ if (~ranks_pos[j]){ low[i] = j; break; } j --; } j = pattern[i] + 1; while(j < s+1){ if (~ranks_pos[j]){ high[i] = j; break; } j ++; } if (ranks_pos[pattern[i]] == -1) ranks_pos[pattern[i]] = i; } } bool check(int A[], int B[], int k, int pos){ if (~ranks_pos[A[pos]]){ if (B[k+ranks_pos[A[pos]]] != B[k+pos]) return false; } if (~low[pos] && ~ranks_pos[low[pos]]){ if (B[k+ranks_pos[low[pos]]] >= B[k+pos]) return false; } if (~high[pos] && ~ranks_pos[high[pos]]){ if (B[k+ranks_pos[high[pos]]] <= B[k+pos]) return false; } return true; } void get_next(){ int j = -1; next[0] = -1; for (int i = 1; i < k; ++ i){ while(j > -1 && !check(pattern, pattern, i-j-1, j+1)) j = next[j]; if (check(pattern, pattern, i-j-1, j+1)) j ++; next[i] = j; //printf("next%d = %d\n", i, next[i]); } } void kmp(){ ans.clear(); get_next(); int j = -1; for (int i = 0; i < n; ++ i){ while(j > -1 && !check(pattern, S, i-j-1, j+1)) j = next[j]; if (check(pattern, S, i-j-1, j+1)) j ++; if (j == k - 1){ ans.push_back(i - k + 1); j = next[j]; } } } int main(){ while(scanf("%d %d %d", &n, &k, &s) != EOF){ for (int i = 0; i < n; i ++) scanf("%d", &S[i]); for (int i = 0; i < k; i ++) scanf("%d", &pattern[i]); init(); kmp(); printf("%d\n", ans.size()); for (int i = 0; i < (int)ans.size(); ++ i){ printf("%d\n", ans[i]+1); } } return 0; } [/cpp]

轉載于:https://www.cnblogs.com/AbandonZHANG/p/4114332.html

總結

以上是生活随笔為你收集整理的POJ 3167 Cow Pattern ★(KMP好题)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。