longest-well-performing-interval_test.go

package problem1124

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

// tcs is testcase slice
var tcs = []struct {
	hours []int
	ans   int
}{

	{
		[]int{9, 6, 6, 6, 9, 9},
		3,
	},

	{
		[]int{6, 9, 6},
		1,
	},

	{
		[]int{13, 8, 13, 4, 10, 11, 12, 5, 3, 0, 14, 16, 9, 2, 11, 12, 11, 10, 6, 14, 8, 6, 3, 6, 2, 4, 3, 3, 10, 5, 10, 13, 0, 5, 16, 6, 15, 12, 5, 3, 5, 9, 7, 2, 6, 14, 12, 12, 16, 13, 1, 7, 2, 6, 9, 5, 16, 5, 4, 15, 0, 12, 4, 13, 8, 10, 0, 4, 15, 5, 3, 2, 11, 6, 7, 5, 12, 6, 4, 15, 16, 1, 1, 11, 13, 10, 16, 2, 11, 15, 0, 1, 10, 5, 15, 16, 9, 6, 9, 10, 5, 0, 6, 11, 12, 12, 6, 14, 13, 3, 8, 16, 16, 10, 16, 0, 9, 2, 5, 5, 7, 11, 8, 5, 15, 6, 16, 14, 13, 4, 7, 7, 1, 12, 3, 10, 1, 8, 10, 0, 12, 15, 14, 0, 3, 1, 5, 15, 5, 8, 4, 8, 12, 16, 10, 6, 11, 8, 5, 11, 6, 11, 15, 15, 5, 11, 1, 11, 9, 8, 13, 10, 9, 3, 7, 7, 6, 6, 15, 1, 15, 15, 2, 9, 6, 0, 6, 10, 3, 1, 4, 8, 4, 3, 0, 15, 10, 10, 9, 3, 12, 0, 13, 6, 2, 12, 11, 14, 9, 9, 1, 15, 10, 3, 7, 0, 11, 3, 8, 1, 9, 5, 14, 9, 12, 7, 8, 12, 14, 0, 3, 14, 12, 6, 1, 14, 0, 1, 0, 8, 12, 2, 2, 6, 0, 4, 4, 11, 11, 3, 7, 1, 3, 11, 4, 14, 4, 11, 12, 1, 13, 4, 3, 6, 9, 7, 7, 10, 15, 3, 2, 16, 14, 6, 14, 13, 8, 7, 0, 10, 4, 15, 3, 0, 12, 14, 0, 1, 14, 13, 14, 12, 7, 1, 9, 10, 3, 7, 8, 7, 0, 12, 15, 10, 3, 2, 3, 4, 6, 0, 4, 4, 13, 13, 0, 14, 12, 1, 13, 13, 16, 11, 3, 15, 5, 0, 2, 10, 6, 14, 16, 11, 11, 11, 12, 8, 3, 16, 7, 13, 2, 1, 13, 14, 1, 9, 8, 5, 14, 4, 8, 2, 12, 7, 7, 11, 2, 15, 15, 12, 12, 9, 9, 13, 8, 0, 12, 8, 2, 14, 2, 6, 1, 3, 3, 0, 0, 6, 0, 16, 10, 10, 9, 4, 4, 5, 12, 0, 3, 16, 11, 10, 0, 16, 15, 14, 2, 0, 1, 6, 15, 7, 12, 3, 1, 5, 2, 0, 11, 16, 2, 16, 1, 14, 7, 12, 16, 1, 5, 5, 6, 4, 3, 3, 12, 8, 14, 0, 2, 14, 3, 7, 9, 1, 8, 16, 12, 4, 13, 13, 9, 14, 12, 6, 16, 16, 9, 4, 15, 0, 2, 13, 4, 7, 10, 8, 2, 10, 13, 5, 9, 16, 10, 4, 15, 1, 10, 8, 7, 5, 3, 7, 6, 14, 1, 5, 3, 15, 2, 13, 2, 12, 13, 13, 11, 5, 6, 11, 9, 14, 11, 5, 8, 10, 0, 0, 12, 0, 7, 4, 0, 9, 12, 14, 13, 6, 7, 1, 14, 5, 5, 2, 5, 11, 5, 14, 3, 6, 9, 14, 12, 7, 7, 6, 0, 12, 1, 4, 1, 12, 6, 4, 0, 2, 16, 3, 4, 1, 12, 15, 16, 14, 3, 15, 7, 10, 7, 10, 14, 3, 8, 14, 5, 9, 3, 9, 0, 7, 7, 2, 1, 14, 16, 10, 2, 16, 4, 2, 3, 9, 2, 8, 11, 2, 10, 13, 9, 0, 4, 0, 16, 6, 9, 9, 14, 1, 8, 10, 5, 12, 9, 10, 14, 0, 5, 3, 7, 15, 11, 9, 8, 2, 16, 6, 10, 12, 2, 0, 13, 16, 4, 11, 8, 2, 3, 15, 11, 4, 16, 1, 2, 0, 15, 15, 11, 14, 10, 8, 14, 2, 11, 11, 14, 10, 2, 6, 11, 0, 15, 15, 3, 0, 6, 9, 10, 7, 11, 15, 16, 14, 10, 15, 6, 10, 4, 1, 1, 11, 10, 3, 8, 8, 11, 6, 8, 2, 15, 11, 16, 2, 15, 4, 5, 5, 13, 12, 1, 9, 14, 9, 10, 7, 7, 9, 10, 9, 6, 12, 12, 1, 14, 16, 13, 5, 5, 8, 8, 14, 0, 16, 15, 8, 10, 9, 15, 4, 12, 15, 5, 4, 12, 5, 9, 9, 1, 6, 12, 12, 1, 8, 2, 15, 1, 11, 1, 1, 1, 1, 10, 14, 3, 10, 12, 5, 6, 12, 16, 12, 6, 7, 0, 9, 12, 8, 5, 11, 5, 11, 5, 11, 6, 14, 1, 0, 13, 0, 14, 6, 7, 13, 5, 14, 1, 13, 16, 12, 8, 15, 13, 16, 16, 2, 14, 3, 16, 2, 1, 14, 2, 3, 9, 3, 10, 5, 5, 1, 2, 2, 3, 8, 0, 16, 14, 11, 3, 8, 16, 16, 16, 13, 16, 15, 15, 11, 13, 15, 7, 12, 5, 5, 10, 7, 1, 3, 15, 10, 16, 14, 16, 2, 4, 0, 12, 1, 5, 5, 15, 11, 6, 12, 1, 8, 10, 1, 4, 0, 15, 10, 16, 6, 5, 1, 0, 7, 12, 15, 12, 5, 0, 2, 0, 9, 10, 6, 9, 5, 7, 0, 12, 15, 8, 11, 16, 12, 12, 3, 2, 14, 5, 16, 7, 8, 15, 0, 16, 16, 6, 3, 7, 13, 14, 5, 9, 12, 9, 9, 0, 0, 7, 11, 8, 4, 5, 16, 10, 5, 0, 9, 0, 12, 9, 2, 3, 4, 11, 5, 14, 11, 1, 7, 12, 2, 2, 13, 11, 11, 2, 2, 12, 7, 12, 9, 16, 11, 14, 13, 14, 15, 13, 9, 10, 12, 12, 7, 14, 7, 10, 0, 6, 10, 1, 1, 13, 16, 11, 12, 13, 3, 10, 7, 3, 16, 2, 5, 8, 2, 0, 7, 14, 8, 1, 11, 13, 9, 4, 2, 13, 13, 11, 11, 14, 15, 12, 4, 3, 12, 15, 14, 12, 2, 4, 3, 11, 12, 8, 15, 7, 11, 15, 15, 6, 8, 14, 13, 0, 14, 7, 5, 10, 4},
		617,
	},

	{
		[]int{6, 6, 6, 6, 9, 9, 9, 9, 9, 9, 6, 0, 6, 6, 9},
		13,
	},

	{
		[]int{16, 15, 11, 2, 9, 3},
		6,
	},

	{
		[]int{9, 9, 6, 0, 6, 6, 9},
		3,
	},

	// 可以有多个 testcase
}

func Test_longestWPI(t *testing.T) {
	ast := assert.New(t)

	for _, tc := range tcs {
		ast.Equal(tc.ans, longestWPI(tc.hours), "输入:%v", tc)
	}
}

func Benchmark_longestWPI(b *testing.B) {
	for i := 0; i < b.N; i++ {
		for _, tc := range tcs {
			longestWPI(tc.hours)
		}
	}
}