algorithms/leetcode/2535-DifferenceBetweenElementSumAndDigitSumOfAnArray.go at master · freewu/algorithms · GitHub

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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package main

// 2535. Difference Between Element Sum and Digit Sum of an Array
// You are given a positive integer array nums.
//     The element sum is the sum of all the elements in nums.
//     The digit sum is the sum of all the digits (not necessarily distinct) that appear in nums.

// Return the absolute difference between the element sum and digit sum of nums.
// Note that the absolute difference between two integers x and y is defined as |x - y|.

// Example 1:
// Input: nums = [1,15,6,3]
// Output: 9
// Explanation:
// The element sum of nums is 1 + 15 + 6 + 3 = 25.
// The digit sum of nums is 1 + 1 + 5 + 6 + 3 = 16.
// The absolute difference between the element sum and digit sum is |25 - 16| = 9.

// Example 2:
// Input: nums = [1,2,3,4]
// Output: 0
// Explanation:
// The element sum of nums is 1 + 2 + 3 + 4 = 10.
// The digit sum of nums is 1 + 2 + 3 + 4 = 10.
// The absolute difference between the element sum and digit sum is |10 - 10| = 0.

// Constraints:
//     1 <= nums.length <= 2000
//     1 <= nums[i] <= 2000

import "fmt"

func differenceOfSum(nums []int) int {
    sum, dsum := 0, 0
    digitSum := func(n int) int {
        res := 0
        for n > 0 {
            res += n % 10
            n /= 10
        }
        return res
    }
    for _, v := range nums {
        sum += v
        dsum += digitSum(v)
    }
    return sum - dsum
}

func differenceOfSum1(nums []int) int {
    sum, dsum := 0, 0
    for _, v := range nums {
        sum += v
        for v > 0 {
            dsum += (v % 10)
            v = v / 10
        }
    }
    return sum - dsum
}


func main() {
    // Example 1:
    // Input: nums = [1,15,6,3]
    // Output: 9
    // Explanation:
    // The element sum of nums is 1 + 15 + 6 + 3 = 25.
    // The digit sum of nums is 1 + 1 + 5 + 6 + 3 = 16.
    // The absolute difference between the element sum and digit sum is |25 - 16| = 9.
    fmt.Println(differenceOfSum([]int{1,15,6,3})) // 9
    // Example 2:
    // Input: nums = [1,2,3,4]
    // Output: 0
    // Explanation:
    // The element sum of nums is 1 + 2 + 3 + 4 = 10.
    // The digit sum of nums is 1 + 2 + 3 + 4 = 10.
    // The absolute difference between the element sum and digit sum is |10 - 10| = 0.
    fmt.Println(differenceOfSum([]int{1,2,3,4})) // 0

    fmt.Println(differenceOfSum1([]int{1,15,6,3})) // 9
    fmt.Println(differenceOfSum1([]int{1,2,3,4})) // 0
}