decrease-elements-to-make-array-zigzag_test.go

package problem1144

import (
	"testing"

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

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

	{
		[]int{875, 303, 979, 293, 693, 803, 81, 17, 329, 262, 669, 10, 413, 142, 537, 82, 857, 283, 492, 779, 118, 216, 728, 920, 490, 258, 225, 109, 252, 292, 517, 653, 885, 680, 12, 2, 399, 286, 365, 234, 731, 927, 409, 325, 339, 416, 713, 946, 551, 122, 794, 308, 518, 505, 365, 285, 481, 167, 107, 861, 357, 299, 588, 669, 622, 520, 983, 728, 10, 995, 250, 790, 346, 34, 482, 551, 641, 376, 646, 561, 861, 311, 849, 231, 627, 810, 848, 840, 602, 872, 813, 818, 462, 990, 545, 936, 870, 745, 1, 654, 473, 517, 160, 245, 36, 445, 149, 968, 313, 488, 468, 625, 479, 256, 993, 127, 224, 558, 40, 450, 587, 400, 254, 830, 707, 300, 501, 86, 408, 448, 956, 658, 799, 669, 154, 226, 531, 907, 551, 713, 353, 245, 883, 683, 368, 777, 677, 597, 166, 226, 74, 208, 882, 940, 271, 586, 730, 346, 572, 742, 325, 293, 263, 617, 962, 781, 817, 73, 320, 41, 484, 849, 919, 266, 419, 685, 211, 240, 223, 596, 859, 662, 337, 14, 803, 476, 763, 62, 115, 491, 840, 623, 593, 22, 763, 145, 616, 115, 393, 51, 239, 264, 968, 998, 391, 49, 159, 297, 410, 807, 787, 510, 860, 769, 703, 834, 298, 249, 229, 960, 98, 849, 65, 704, 340, 60, 174, 263, 261, 38, 79, 661, 237, 853, 930, 205, 602, 498, 117, 461, 918, 540, 261, 71, 440, 476, 801, 455, 235, 346, 673, 385, 205, 379, 238, 443, 861, 844, 718, 451, 492, 708, 528, 219, 749, 500, 67, 219, 489, 966, 165, 403, 359, 684, 961, 547, 832, 678, 674, 446, 831, 804, 479, 696, 471, 164, 577, 482, 237, 557, 123, 669, 940, 809, 546, 120, 973, 835, 549, 958, 960, 494, 369, 496, 600, 594, 485, 517, 403, 174, 308, 705, 431, 644, 441, 220, 836, 919, 634, 542, 564, 763, 481, 993, 603, 300, 379, 990, 377, 631, 147, 185, 997, 244, 972, 482, 410, 97, 167, 103, 235, 316, 149, 894, 399, 706, 651, 881, 185, 683, 479, 791, 5, 161, 746, 34, 735, 151, 405, 616, 478, 372, 831, 798, 918, 218, 752, 368, 208, 852, 582, 496, 281, 789, 339, 887, 90, 447, 40, 156, 163, 242, 276, 159, 175, 122, 3, 403, 327, 168, 353, 634, 935, 144, 411, 321, 956, 931, 632, 548, 710, 932, 366, 820, 46, 113, 780, 637, 738, 64, 253, 129, 279, 885, 476, 493, 712, 410, 325, 85, 435, 552, 545, 936, 247, 598, 615, 546, 637, 582, 300, 571, 425, 180, 446, 552, 509, 109, 653, 404, 739, 387, 650, 535, 247, 590, 628, 455, 993, 19, 108, 513, 316, 975, 534, 375, 850, 911, 403, 476, 713, 426, 904, 934, 8, 306, 41, 517, 491, 744, 720, 517, 589, 21, 354, 954, 63, 922, 863, 687, 376, 421, 185, 988, 643, 96, 724, 210, 218, 612, 207, 437, 927, 728, 666, 598, 142, 670, 165, 561, 942, 447, 916, 363, 731, 441, 318, 613, 750, 108, 176, 472, 85, 855, 144, 635, 78, 115, 586, 209, 845, 149, 905, 891, 390, 921, 741, 874, 691, 212, 784, 580, 364, 324, 907, 830, 669, 204, 31, 340, 830, 7, 15, 742, 17, 197, 223, 615, 335, 684, 172, 685, 193, 44, 19, 669, 446, 525, 361, 591, 847, 900, 734, 928, 56, 939, 412, 488, 732, 186, 816, 165, 77, 564, 147, 376, 712, 82, 128, 726, 555, 72, 316, 983, 330, 219, 339, 30, 283, 557, 879, 774, 7, 372, 215, 106, 322, 207, 946, 389, 333, 348, 969, 730, 899, 95, 301, 489, 407, 688, 929, 438, 672, 513, 210, 533, 38, 604, 160, 541, 80, 932, 871, 510, 78, 450, 884, 306, 867, 466, 530, 766, 220, 907, 736, 571, 70, 380, 700, 216, 482, 414, 670, 47, 481, 389, 634, 199, 178, 455, 90, 579, 484, 988, 38, 160, 545, 21, 234, 712, 221, 813, 566, 847, 605, 183, 338, 513, 304, 667, 982, 426, 335, 982, 420, 956, 809, 809, 876, 118, 577, 276, 965, 944, 917, 383, 315, 518, 96, 376, 648, 247, 831, 61, 872, 146, 572, 446, 203, 243, 86, 791, 418, 629, 485, 306, 411, 539, 83, 126, 589, 119, 545, 580, 627, 15, 909, 135, 539, 15, 764, 58, 120, 595, 846, 687, 397, 969, 144, 179, 308, 867, 569, 202, 492, 581, 130, 111, 630, 966, 378, 463, 349, 589, 241, 423, 774, 54, 639, 594, 492, 974, 766, 744, 540, 301, 565, 139, 123, 110, 216, 539, 480, 656, 647, 63, 369, 847, 546, 623, 584, 786, 407, 289, 361, 792, 329, 782, 686, 132, 98, 343, 582, 999, 427, 310, 376, 254, 980, 397, 912, 975, 201, 636, 538, 883, 245, 505, 493, 894, 766, 772, 340, 458, 112, 128, 803, 395, 989, 558, 296, 94, 837, 653, 360, 866, 325, 675, 435, 93, 127, 199, 531, 165, 892, 80, 648, 496, 127, 253, 959, 703, 912, 216, 228, 615, 412, 868, 212, 768, 626, 827, 653, 77, 998, 148, 593, 745, 807, 917, 468, 289, 909, 267, 639, 649, 743, 527, 26, 591, 857, 997, 860, 83, 736, 412, 9, 293, 872, 792, 248, 755, 380, 401, 804, 15, 229, 196, 350, 760, 1, 433, 541, 626, 359, 271, 426, 20, 548, 24, 377, 326, 583, 89, 376, 342, 831, 629, 349, 317, 377, 401, 369, 852, 930, 286, 200, 170, 332, 316, 427, 425, 377, 325, 987, 599, 463, 127, 348, 519, 754, 361, 917, 176, 821, 7, 775, 346, 597, 369, 362, 646, 19, 926, 976, 28, 912, 209, 378, 671, 401, 627, 754, 50, 11, 143, 554, 810, 896, 586, 235, 293, 379, 628, 202, 606, 930, 807, 707, 530, 602, 219, 246, 229, 596, 481, 789, 429, 778, 379, 684, 474, 210, 2, 19, 874, 357, 247, 259, 334, 615, 667, 534, 578, 178, 743, 953, 399, 583, 689, 858, 536, 829, 661, 402, 401, 477, 859, 257, 92},
		113084,
	},

	{
		[]int{1, 2, 3},
		2,
	},

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

	// 可以有多个 testcase
}

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

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

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