猿辅导笔试

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;
class Solution {
public:
	int findAnagrams(vector<int>& nums, int k) {
		
        sort(nums.begin(), nums.end());
        int res = 0;
		//[l-p.size()+1,l]区间内判断是或否与p anagram 有k个元素
		for (int r = 0; r < nums.size(); r++) {
			int l = r - k + 1;
            if(l>=0 && nums[l]<=0)
                continue;
            //长度等于k，这样下一轮就有p.size()个值
			if (l >= 0){
                int js = nums[l];
                res += js;
                for(int i=l; i<=l+2; i++){
                    nums[i] -= js;
                }   
            }
            // cout << r <<":" <<endl;
            // cout << l <<" "<<endl;
            // printVec (nums);
		}
		return res;
	}
};

int main() {
    int c;
    cin >> c;
    int n, num;
    for(int i=0; i<c; i++){
        vector<int> nums;
        cin >> n;
        for(int j=0; j<n; j++){
            cin >> num;
            nums.push_back(num);
        }
        int res = Solution().findAnagrams(nums, 3);
        cout << res <<endl;
    }
	return 0;
}