LeetCode 1395.统计作战单位数

n 名士兵站成一排。每个士兵都有一个 独一无二 的评分 rating 。

每 3 个士兵可以组成一个作战单位，分组规则如下：

从队伍中选出下标分别为 i、j、k 的 3 名士兵，他们的评分分别为 rating[i]、rating[j]、rating[k]
作战单位需满足： rating[i] < rating[j] < rating[k] 或者 rating[i] > rating[j] > rating[k] ，其中 0 <= i < j < k < n
请你返回按上述条件可以组建的作战单位数量。每个士兵都可以是多个作战单位的一部分。

示例 1：

输入：rating = [2,5,3,4,1]
输出：3
解释：我们可以组建三个作战单位 (2,3,4)、(5,4,1)、(5,3,1) 。
示例 2：

输入：rating = [2,1,3]
输出：0
解释：根据题目条件，我们无法组建作战单位。
示例 3：

输入：rating = [1,2,3,4]
输出：4

提示：

n == rating.length
1 <= n <= 200
1 <= rating[i] <= 10^5

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-number-of-teams
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

package leetcode周赛;
//找出从大到小的组合呗或者从小到大
//从第一个数开始与后面的数判断
//将题中数组复制3份，寻找第二份中比第一份小的数，且要保证i的小于j
//所以要同时记录j和值，第三份中j < k且比第二份小的，满足则加一，大的同理
//于是过程：
//1、开辟两个for循环，两个bmin数组二维数组，bmax
//2、for三的满足，得到结果
public class _04统计作战单位数 {

	public static void main(String[] args) {
		int[] rating = {3,6,7,5,1};
		System.out.println(numTeams(rating));
	}
	 public static int numTeams(int[] rating) {
	     
	     //其实只需要存入索引就ok了吧
	     //判断3的时候，将bmin[j]=1；当作记录
		 int bmin[] = new int[205];
		 int bmax[] = new int[205];
		 for(int i = 0; i < rating.length-1; i++) {
			 for(int j = i+1; j < rating.length; j++) {//指的第二个数
				 if(rating[i] < rating[j]) {
					 bmin[j]++;
				 }
				 if(rating[i] > rating[j]) {
					 bmax[j]++;
				 }
			 }
		 }
		 int res = 0;
		 for(int i =0; i < rating.length-1; i++) {
			 for(int j = i+1; j < rating.length; j++) {//指的第三个数
				 if(rating[i] < rating[j]) {//意思是：第三数也小于第二个数，且第二个数的也有满足的第一个数
					 if(bmin[i] != 0) {
						 res += bmin[i];
					 }
				 }
				 if(rating[i] > rating[j]) {
					 if(bmax[i] != 0) {
						 res += bmax[i];
					 }
				 }
			 }
		 }
		 return res;
	  }
}