summaryrefslogtreecommitdiff
path: root/src/bin/p0005.rs
blob: ce4d5e49cedb72ac12545b0b930eb2f0c64292f0 (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
impl Solution {
    pub fn check_palindrome_len(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 len1 = 2 * r + 1;
        let start1 = i - r;
        let end1 = i + r;

        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 len2 = 2 * r;
        let start2 = i - r + 1;
        let end2 = i + r;

        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()));
}