1395 - Count Number of Teams

 JAVA

  • class Solution {
        public int numTeams(int[] rating) {
            int n = rating.length;
            int ans = 0;
            for (int i = 0; i < n; ++i) {
                int l = 0, r = 0;
                for (int j = 0; j < i; ++j) {
                    if (rating[j] < rating[i]) {
                        ++l;
                    }
                }
                for (int j = i + 1; j < n; ++j) {
                    if (rating[j] > rating[i]) {
                        ++r;
                    }
                }
                ans += l * r;
                ans += (i - l) * (n - i - 1 - r);
            }
            return ans;
        }
    }

C++

  • class Solution {
    public:
        int numTeams(vector<int>& rating) {
            int n = rating.size();
            int ans = 0;
            for (int i = 0; i < n; ++i) {
                int l = 0, r = 0;
                for (int j = 0; j < i; ++j) {
                    if (rating[j] < rating[i]) {
                        ++l;
                    }
                }
                for (int j = i + 1; j < n; ++j) {
                    if (rating[j] > rating[i]) {
                        ++r;
                    }
                }
                ans += l * r;
                ans += (i - l) * (n - i - 1 - r);
            }
            return ans;
        }
    };

Comments