summaryrefslogtreecommitdiff
path: root/rust/src/bin/p0005.rs
blob: 01b368d24f79629426857afd315ed46bce6c9c2d (plain)
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
impl Solution {
    fn check_odd(s: &String, i: i32) -> (i32, i32, i32) {
        let mut r = 0;
        loop {
            let next = r + 1;
            if i - next < 0
                    || i + next >= s.len() as i32
                    || s.as_bytes()[(i-next) as usize] != s.as_bytes()[(i+next) as usize] {
                break;
            }
            r = next;
        }
        let len = 2 * r + 1;
        let start = i - r;
        let end = i + r;
        (len, start, end)
    }

    fn check_even(s: &String, i: i32) -> (i32, i32, i32) {
        let mut r = 0;
        loop {
            let next = r + 1;
            if i - next + 1 < 0
                || i + next >= s.len() as i32
                || s.as_bytes()[(i-next+1) as usize] != s.as_bytes()[(i+next) as usize] {
                break;
            }
            r = next;
        }
        let len = 2 * r;
        let start = i - r + 1;
        let end = i + r;
        (len, start, end)
    }

    fn check_palindrome_len(s: &String, i: i32) -> (i32, i32, i32) {
        let (len1, start1, end1) = Self::check_odd(s, i);
        let (len2, start2, end2) = Self::check_even(s, i);
        if len2 > len1 {
            (len2, start2, end2)
        } else {
            (len1, start1, end1)
        }
    }

    pub fn longest_palindrome(s: String) -> String {
        let mut maxpalinlen = -1;
        let mut start: i32 = 0;
        let mut end: i32 = 0;
        for i in 0..s.len() {
            let (palin_len, pstart, pend) = Self::check_palindrome_len(&s, i as i32);
            if palin_len > maxpalinlen {
                maxpalinlen = palin_len;
                start = pstart;
                end = pend;
            }
        }
        s[start as usize ..(end + 1) as usize].to_string()
    }
}

struct Solution {}

fn main () {
    println!("{}", Solution::longest_palindrome("babad".to_string()));
    println!("{}", Solution::longest_palindrome("cbbd".to_string()));
}