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<type.length; i++) {
if (type[i] <= type.length) {
c[type[i]-1] += 1;
}
}

for (int i=0; i<c.length; 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颇为接近。