Leecode 第 77 场双周赛 6052. 最小平均差

原题链接：Leecode 第 77 场双周赛 6052. 最小平均差

class Solution {
public:
    static bool cmp(const pair<int ,long long>& a, const pair<int,long long>& b)
    {
        if(a.second!=b.second) return a.second<b.second;
        else return a.first<b.first;
    }
    int minimumAverageDifference(vector<int>& nums) {
        int n=nums.size();
        long long  a[n];
        a[0]=nums[0];
        vector<pair<int,long long> > v;
        for(int i=1;i<n;i++)
        {
            a[i]=a[i-1]+nums[i];
        }
        for(int i=0;i<n;i++)
        {
            long long l=a[i]/(i+1);
            long long r=n-i-1==0 ? 0 : (a[n-1]-a[i])/(n-i-1);
            long long num=abs(l-r);
            v.push_back(make_pair(i,num));
        }
        sort(v.begin(),v.end(),cmp);
        return v[0].first;
    }
};