码迷,mamicode.com
首页 > 其他好文 > 详细

1395. Count Number of Teams

时间:2020-07-18 16:04:17      阅读:67      评论:0      收藏:0      [点我收藏+]

标签:size   xpl   三元组   span   can   vector   hang   class   tput   

问题:

给定一组带有评分rating的士兵序列。

求从中挑出3个士兵 i, j, k (i<j<k)为一个小组,使得第 i, j, k 名士兵的 rating递增or递减。

这样的小组有多少个。

Example 1:
Input: rating = [2,5,3,4,1]
Output: 3
Explanation: We can form three teams given the conditions. (2,3,4), (5,4,1), (5,3,1). 

Example 2:
Input: rating = [2,1,3]
Output: 0
Explanation: We can‘t form any team given the conditions.

Example 3:
Input: rating = [1,2,3,4]
Output: 4
 
Constraints:
n == rating.length
1 <= n <= 200
1 <= rating[i] <= 10^5

  

解法:

由于是3元组,

那么我们以中间元素 j 为参照系,

分别 在 j 的左边(i<j) 找 i,在 j 的右边(j<k) 找 k

1. 递增的情况:

找到 i 满足 rating[i]<rating[j]:有x1种情况
找到 k 满足 rating[j]<rating[k]:有y1种情况

则满足题意的三元组有 x1*y1 种情况。

2. 递减的情况:

找到 i 满足 rating[i]>rating[j]:有x2种情况
找到 k 满足 rating[j>rating[k]:有y2种情况

则满足题意的三元组有 x2*y2 种情况。

对此时的 j 来说,一共有 x1*y1+x2*y2 种情况。

 

代码实现上,

我们遍历 j 从第二个元素 j=1 开始,到倒数第二个元素 j=size-2。(由于j为三元组中间元素,前面必有 i,后面必有 k)

对于每一个 j ,

我们遍历整个序列,找 i,k

使用 less保存 rating i < rating j 的情况

使用 greater保存  rating i > rating j 的情况

那么同时找 i 和 k 的话,一个 小于 j, 一个 大于 j

less[i<j] -> less[0]:i>j: i代表k && rating i < rating j 的情况
                less[1]: i<j: i代表i && rating i < rating j 的情况
greater[i<j] -> greater[0]:i>j: i代表k && rating i > rating j 的情况
                     greater[1]: i<j: i代表i && rating i > rating j 的情况

因此,对于当前的 j 来说,有

less[0]*greater[1]  : i<j<k && rating[i] > rating[j] > rating[k] :递减

+

less[1]*greater[0] : i<j<k && rating[i] < rating[j] < rating[k] :递增

种情况。

 

代码参考:

 1 class Solution {
 2 public:
 3     int numTeams(vector<int>& rating) {
 4         int res=0;
 5         for(int j=1; j<rating.size()-1; j++){
 6             int less[2]={0}, greater[2]={0};
 7             for(int i=0; i<rating.size(); i++){
 8                 if(rating[i]<rating[j]){
 9                     less[i<j]++;
10                 }else if(rating[i]>rating[j]){
11                     greater[i<j]++;
12                 }
13             }
14             res += (less[1]*greater[0] + less[0]*greater[1]);
15         }
16         return res;
17     }
18 };

 

1395. Count Number of Teams

标签:size   xpl   三元组   span   can   vector   hang   class   tput   

原文地址:https://www.cnblogs.com/habibah-chang/p/13335718.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!