LinkedIn 实习电面 ml track

zj00377 发表于 2015-12-6 15:15
对呀，看了面我的人和 manager 的主页, 里面优化的方法都是ADMM 来解 logistic regression.

借LZ吉言, 我拿到onsite了！

UltraInstinct 发表于 2015-12-6 15:54
借LZ吉言, 我拿到onsite了！

哈哈，祝你好运！

这题可以用backtracking求解，贴个我的java代码，返回值是所有可能的index组合~

1. public List<int[]> getAllIndex (int[] dimensions) {
   
2.                 List<int[]> result = new ArrayList<int[]>();
   
3.                 int[] index = new int[dimensions.length];
   
4.                 helper(result, index, 0, dimensions);
   
5.                 return result;
   
6.         }
   
7.        
   
8.         private void helper(List<int[]> result, int[] index, int depth, int[] dimensions) {
   
9.                 if (depth == dimensions.length) {
   
10.                         result.add(index.clone());
    
11.                         return;
    
12.                 }
    
13.                 int currentDimension = dimensions[depth];
    
14.                 for (int i=0; i<currentDimension; i++) {
    
15.                         index[depth] = i;
    
16.                         helper(result, index, depth+1, dimensions);
    
17.                 }
    
18.         }

复制代码

罹年 发表于 2015-12-8 13:35
这题可以用backtracking求解，贴个我的java代码，返回值是所有可能的index组合~

这个写的好

贴一下python code

1. def helper(nums, res, temp, n):
   
2.     if len(temp) == n:
   
3.         res.append(temp[:])
   
4.         return
   
5.     for i in range(nums[0]):
   
6.         temp.append(i)
   
7.         helper(nums[1:], res, temp, n)
   
8.         temp.pop()
   
9. 
   
10. def getsum(data, dims):
    
11.     index = []
    
12.     helper(dims, index, [], len(dims))
    
13.     total = 0
    
14.     for idx in index:
    
15.         total += getelement(data, idx)
    
16. 
    
17.     return total
    
18. 
    
19. def getelement(data, idx):
    
20.     if not idx:
    
21.         return data
    
22.     n = idx.pop(0)
    
23.     return getelement(data[n], idx)
    
24. 
    
25. data = [ [[1,2,3], [0,3,4]], [[1,2,3], [0,3,4]]]
    
26. dims = [2,2,3]
    
27. print(getsum(data, dims))

复制代码