Count Arrays with Same Digit Sum

Given an integer array, return the number of subarrays where each number has the same digit sum. For example, for the array [4998, 4999, 4999], return the total number of valid subarrays. Example Input [4998, 4999, 4999]