2535. Difference Between Element Sum and Digit Sum of an Array

class Solution {
    public int differenceOfSum(int[] nums) {
        int elementSum = 0;
        int digitSum = 0;
        
        for (int num : nums) {
            elementSum += num;
            
            while (num > 0) {
                digitSum += num % 10;
                num /= 10;
            }
        }
        
        if (elementSum > digitSum) {
            return elementSum - digitSum;
        } else {
            return digitSum - elementSum;
        }
    }
}