mirror of
https://hub.njuu.cf/TheAlgorithms/C-Plus-Plus.git
synced 2023-10-11 13:05:55 +08:00
Updated int -> uint64_t
Updated int -> uint64_t for non-negative values
This commit is contained in:
parent
edff958ab2
commit
12d5123995
@ -26,11 +26,11 @@
|
|||||||
* \param[in] pattern text on which to apply the Z-function
|
* \param[in] pattern text on which to apply the Z-function
|
||||||
* \returns the Z-function output as a vector array
|
* \returns the Z-function output as a vector array
|
||||||
*/
|
*/
|
||||||
std::vector<int> Z_function(const std::string &pattern) {
|
std::vector<uint64_t> Z_function(const std::string &pattern) {
|
||||||
int pattern_length = pattern.size();
|
uint64_t pattern_length = pattern.size();
|
||||||
std::vector<int> z(pattern_length, 0);
|
std::vector<uint64_t> z(pattern_length, 0);
|
||||||
|
|
||||||
for (int i = 1, l = 0, r = 0; i < pattern_length; i++) {
|
for (uint64_t i = 1, l = 0, r = 0; i < pattern_length; i++) {
|
||||||
if (i <= r)
|
if (i <= r)
|
||||||
z[i] = std::min(r - i + 1, z[i - l]);
|
z[i] = std::min(r - i + 1, z[i - l]);
|
||||||
while (i + z[i] < pattern_length && pattern[z[i]] == pattern[i + z[i]])
|
while (i + z[i] < pattern_length && pattern[z[i]] == pattern[i + z[i]])
|
||||||
@ -47,13 +47,13 @@ std::vector<int> Z_function(const std::string &pattern) {
|
|||||||
* \param[in] text text in which to search
|
* \param[in] text text in which to search
|
||||||
* \returns a vector of starting indexes where pattern is found in the text
|
* \returns a vector of starting indexes where pattern is found in the text
|
||||||
*/
|
*/
|
||||||
std::vector<int> find_pat_in_text(const std::string &pattern,
|
std::vector<uint64_t> find_pat_in_text(const std::string &pattern,
|
||||||
const std::string &text) {
|
const std::string &text) {
|
||||||
int text_length = text.size(), pattern_length = pattern.size();
|
uint64_t text_length = text.size(), pattern_length = pattern.size();
|
||||||
std::vector<int> z = Z_function(pattern + '#' + text);
|
std::vector<uint64_t> z = Z_function(pattern + '#' + text);
|
||||||
std::vector<int> matching_indexes;
|
std::vector<uint64_t> matching_indexes;
|
||||||
|
|
||||||
for (int i = 0; i < text_length; i++) {
|
for (uint64_t i = 0; i < text_length; i++) {
|
||||||
if (z[i + pattern_length + 1] == pattern_length)
|
if (z[i + pattern_length + 1] == pattern_length)
|
||||||
matching_indexes.push_back(i);
|
matching_indexes.push_back(i);
|
||||||
}
|
}
|
||||||
@ -69,15 +69,15 @@ static void test() {
|
|||||||
std::string text1 = "alskfjaldsabc1abc1abcbksbcdnsdabcabc";
|
std::string text1 = "alskfjaldsabc1abc1abcbksbcdnsdabcabc";
|
||||||
std::string pattern1 = "abc";
|
std::string pattern1 = "abc";
|
||||||
|
|
||||||
std::vector<int> matching_indexes1 = find_pat_in_text(pattern1, text1);
|
std::vector<uint64_t> matching_indexes1 = find_pat_in_text(pattern1, text1);
|
||||||
assert((matching_indexes1 == std::vector<int>{10, 14, 18, 30, 33}));
|
assert((matching_indexes1 == std::vector<uint64_t>{10, 14, 18, 30, 33}));
|
||||||
|
|
||||||
// corner case
|
// corner case
|
||||||
std::string text2 = "greengrass";
|
std::string text2 = "greengrass";
|
||||||
std::string pattern2 = "abc";
|
std::string pattern2 = "abc";
|
||||||
|
|
||||||
std::vector<int> matching_indexes2 = find_pat_in_text(pattern2, text2);
|
std::vector<uint64_t> matching_indexes2 = find_pat_in_text(pattern2, text2);
|
||||||
assert((matching_indexes2 == std::vector<int>{}));
|
assert((matching_indexes2 == std::vector<uint64_t>{}));
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
Loading…
Reference in New Issue
Block a user