Advertisement
bayareabelletrist

Nested array sum

Jul 21st, 2018
130
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. @interface Solution: NSObject
  2. -(int)arrayValue:(NSArray *)arr;
  3. @end
  4.  
  5. @implementation Solution
  6. -(int)arrayValue:(NSArray *)arr {
  7.     return [self pfx_arrayValue:arr depth:1];
  8. }
  9.  
  10. -(int)pfx_arrayValue:(NSArray *)arr depth:(int)d {
  11.     int sum = 0;
  12.     for(id element in arr) {
  13.         if([element isKindOfClass:[NSNumber class]]) {
  14.             NSNumber *num = (NSNumber *)element;
  15.             sum += num.intValue * d;
  16.         } else {
  17.             sum += [self pfx_arrayValue:(NSArray *)element depth:d+1];
  18.         }
  19.     }
  20.     return sum;
  21. }
  22. @end
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement