糖逗 发表于 2020-6-14 20:18:11

C++刷leetcode(1395. 统计作战单位数)【动态规划】

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

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

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

 

示例 1:

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

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

输入:rating =
输出:4
 

提示:

n == rating.length
1 <= n <= 200
1 <= rating <= 10^5

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


class Solution {
public:
    int numTeams(vector<int>& rating) {
      int len = rating.size();
      vector<vector<vector<int> > > dp(len, vector<vector<int> >(3, vector<int>(2, 0)));
      for(int i = 0; i < len;i ++){
            dp = 1;
            dp = 1;
      }
      for(int i = 0; i < len; i++){
            for(int j = 0; j < i; j++){
                if(rating < rating){
                  for(int k = 1; k < 3; k++){
                        dp += dp;
                  }
                }else if(rating > rating){
                  for(int k = 1; k < 3; k++){
                        dp += dp;
                  }
                }
            }
      }
      int res = 0;
      for(int i = 0; i < len; i++){
            res += dp;
            res += dp;
      }
      return res;
    }
};


注意事项:
1. dp表示以rating为结尾的j长度递增或递减的总数(k=0表示递减,k=1表示递增)

糖逗 发表于 2020-6-14 20:19:03

{:10_327:}
页: [1]
查看完整版本: C++刷leetcode(1395. 统计作战单位数)【动态规划】