JAVA中getlngo_GetNumber的实现（Python & Java & Golang）

TCO2014的编程赢取门票的题目，大致是从一个数组(大小为K)，可以选取1～K个数，必须保证这n个数是从1～n，返回所有的选取方法个数。

思路：首先是得到从1开始连续的数，保存每个数的个数。然后通过排列组合得到结果。

当时采用的Python编写，大致的模样是这样(题目要求的类名就略去了)：

def getNumber(type):

c = [0] * len(type)

s = 0

p = 0

for x in type:

if x <= len(type):

c[x-1] += 1

for i in c:

if i == 0:

return s

else:

if s == 0:

s = i

p = i

else:

p *= i

s += p

return s

print getNumber([2])

print getNumber([1, 2])

print getNumber([1, 3, 2])

print getNumber([1, 1, 2])

print getNumber([1, 3, 2, 5, 7, 4, 5, 4])

print getNumber([1, 1, 2, 2, 3, 3, 4, 4, 5, 5])

之后尝试了一下 Java 版本：

package org.huys.algo.problem;

public class WinterAndCandies {

public static int getNumber(int[] type) {

int[] c = new int[type.length];

int s = 0;

int p = 0;

for (int i=0; i

if (type[i] <= type.length) {

c[type[i]-1] += 1;

}

}

for (int i=0; i

if (c[i] == 0) {

return s;

} else {

if (s == 0) {

s = c[i];

p = c[i];

} else {

p *= c[i];

s += p;

}

}

}

return s;

}

public static void main(String[] args) {

System.out.println(getNumber(new int[] {2}));

System.out.println(getNumber(new int[] {1, 2}));

System.out.println(getNumber(new int[] {1, 3, 2}));

System.out.println(getNumber(new int[] {1, 1, 2}));

System.out.println(getNumber(new int[] {1, 3, 2, 5, 7, 4, 5, 4}));

System.out.println(getNumber(new int[] {1, 1, 2, 2, 3, 3, 4, 4, 5, 5}));

}

}

最近开始关注golang和swift，两种语言有很多共通之处。golang没有class，还是不免有些不适。

package main

import "fmt"

func getNumber(types []int) int {

var s = 0

var p = 0

var c = make([]int, len(types))

for _, value := range types {

if value <= len(types) {

c[value-1] += 1

}

}

for _, i := range c {

if i == 0 {

return s

} else {

if s == 0 {

s = i

p = i

} else {

p *= i

s += p

}

}

}

return s

}

func main() {

fmt.Println(getNumber([]int {2}))

fmt.Println(getNumber([]int {1, 2}))

fmt.Println(getNumber([]int {1, 3, 2}))

fmt.Println(getNumber([]int {1, 1, 2}))

fmt.Println(getNumber([]int {1, 3, 2, 5, 7, 4, 5, 4}))

fmt.Println(getNumber([]int {1, 1, 2, 2, 3, 3, 4, 4, 5, 5}))

}

代码中主要是数组相关操作。相对而言，Python 最灵活。Golang 中采用 Slice 语法，与同为C系语言的Java颇为接近。