您的位置:首页 > 编程语言 > Go语言

算法基础:田忌赛马问题(Golang实现)

2015-08-14 17:15 896 查看
【田忌赛马问题】

输入:

输入有多组测试数据。 每组测试数据包括3行:

第一行输入N(1≤N≤1000),表示马的数量。

第二行有N个整型数字,即渊子的N匹马的速度(数字大表示速度快)。

第三行有N个整型数字,即对手的N匹马的速度。

当N为0时退出。

输出:

若通过聪明的你精心安排,如果能赢得比赛(赢的次数大于比赛总次数的一半),那么输出“YES”。 否则输出“NO”。

样例输入

5

2 3 3 4 5

1 2 3 4 5

4

2 2 1 2

2 2 3 1

0

样例输出

YES

NO

代码实现(Golang):

package huawei

//Date:2015-8-14 15:43:11
import (
"fmt"
"io/ioutil"
"sort"
"strings"
)

//思路:用自己最强的(半数+1)个马和对手最弱的(半数+1)个马比赛
func Test11Base() {
data, err := ioutil.ReadFile("DataFiles/huawei_test11.txt")
checkError(err, "Reading file")

strs := strings.Split(string(data), "\n")
index := 0
for {
count := strs[index]
if count == "0" {
break
}

teamA := convertToIntSlice(strings.Fields(strs[index+1]))
teamB := convertToIntSlice(strings.Fields(strs[index+2]))

if canWin(teamA, teamB) {
fmt.Println("YES")
} else {
fmt.Println("NO")
}

index += 3
}
}

//判断teamA是否能够胜利
func canWin(teamA []int, teamB []int) bool {
sort.Ints(teamA)
sort.Ints(teamB)

length := len(teamA)
tryCount := length/2 + 1
for i := 0; i < tryCount; i++ {
//A组最强的一半
speedA := teamA[length-(tryCount-i)]
//B组最弱的一半
speedB := teamB[i]

if speedA <= speedB {
return false
}
}

return true
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  golang