Skip to content

Instantly share code, notes, and snippets.

@NodoFox
Created January 6, 2017 05:52
Show Gist options
  • Save NodoFox/c1fb427b73322e9c8e6a34c018879ff1 to your computer and use it in GitHub Desktop.
Save NodoFox/c1fb427b73322e9c8e6a34c018879ff1 to your computer and use it in GitHub Desktop.
Sum of nested list of 'NestedInteger'
public class Solution {
public long sum(List<NestedInteger> list) {
return sumRecurse(list, 0); // 0 being the depth
}
public long sumRecurse(List<NestedInteger> list, int level) {
if(list == null || list.size() == 0) {
return 0;
}
long sum = 0;
for(NestedInteger element : list) {
if(element.isInteger()) {
sum = sum + element.getInteger() * level;
}else {
sum = sum + sumRecurse(element.getList(), level + 1);
}
}
return sum;
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment