2544. 交替数字和
题目描述
给你一个正整数 n
。n
中的每一位数字都会按下述规则分配一个符号:
- 最高有效位 上的数字分配到 正 号。
- 剩余每位上数字的符号都与其相邻数字相反。
返回所有数字及其对应符号的和。
示例 1:
输入:n = 521 输出:4 解释:(+5) + (-2) + (+1) = 4
示例 2:
输入:n = 111 输出:1 解释:(+1) + (-1) + (+1) = 1
示例 3:
输入:n = 886996 输出:0 解释:(+8) + (-8) + (+6) + (-9) + (+9) + (-6) = 0
提示:
1 <= n <= 109
解法
方法一:模拟
直接根据题目描述模拟即可。
我们定义一个初始符号 $sign=1$,然后从最高有效位开始,每次取出一位数字 $x$,与 $sign$ 相乘,将结果加到答案中,然后将 $sign$ 取反,继续处理下一位数字,直到处理完所有数字。
时间复杂度 $O(\log n)$,空间复杂度 $O(\log n)$。其中 $n$ 为给定数字。
Python3
class Solution:
def alternateDigitSum(self, n: int) -> int:
return sum((-1) ** i * int(x) for i, x in enumerate(str(n)))
Java
class Solution {
public int alternateDigitSum(int n) {
int ans = 0, sign = 1;
for (char c : String.valueOf(n).toCharArray()) {
int x = c - '0';
ans += sign * x;
sign *= -1;
}
return ans;
}
}
C++
class Solution {
public:
int alternateDigitSum(int n) {
int ans = 0, sign = 1;
for (char c : to_string(n)) {
int x = c - '0';
ans += sign * x;
sign *= -1;
}
return ans;
}
};
Go
func alternateDigitSum(n int) (ans int) {
sign := 1
for _, c := range strconv.Itoa(n) {
x := int(c - '0')
ans += sign * x
sign *= -1
}
return
}
TypeScript
function alternateDigitSum(n: number): number {
let ans = 0;
let sign = 1;
while (n) {
ans += (n % 10) * sign;
sign = -sign;
n = Math.floor(n / 10);
}
return ans * -sign;
}
Rust
impl Solution {
pub fn alternate_digit_sum(mut n: i32) -> i32 {
let mut ans = 0;
let mut sign = 1;
while n != 0 {
ans += (n % 10) * sign;
sign = -sign;
n /= 10;
}
ans * -sign
}
}
C
int alternateDigitSum(int n) {
int ans = 0;
int sign = 1;
while (n) {
ans += (n % 10) * sign;
sign = -sign;
n /= 10;
}
return ans * -sign;
}
方法二
Python3
class Solution:
def alternateDigitSum(self, n: int) -> int:
ans, sign = 0, 1
for c in str(n):
x = int(c)
ans += sign * x
sign *= -1
return ans
Rust
impl Solution {
pub fn alternate_digit_sum(n: i32) -> i32 {
let mut ans = 0;
let mut sign = 1;
for c in format!("{}", n).chars() {
let x = c.to_digit(10).unwrap() as i32;
ans += x * sign;
sign *= -1;
}
ans
}
}