每天一道Rust-LeetCode(2019-06-01)
坚持每天一道题,刷题学习Rust.
题目描述
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
解题过程
impl Solution {
pub fn add_two_numbers(
l1: Option<Box<ListNode>>,
l2: Option<Box<ListNode>>,
) -> Option<Box<ListNode>> {
let (mut p, mut q, mut carry) = (l1, l2, 0);
let mut head = Some(Box::new(ListNode::new(0)));
let mut current = head.as_mut();
while p.is_some() || q.is_some() {
let mut sum = carry;
if let Some(v) = p {
sum += v.val;
p = v.next;
}
if let Some(v) = q {
sum += v.val;
q = v.next;
}
carry = sum / 10;
let v = current.unwrap();
v.next = Some(Box::new(ListNode::new(sum % 10)));
current = v.next.as_mut();
}
if carry > 0 {
let v = current.unwrap();
v.next = Some(Box::new(ListNode::new(carry)));
}
head.unwrap().next
}
}
一点感悟
rust语言和以前接触到的c++,java,go这些最大的区别就是没有零值. 因为这个原因,所有它提供了Option,应该是习惯的原因,用起来是相当大不顺手.
这个题难点就是在熟悉rust的用法,算法本身思路很清晰.
其他
欢迎关注我的github,本项目文章所有代码都可以找到.