是一道好题
设s[0]=0,sum[i]=s[0]+s[1]+...s[i]
那么对于一个可行区间[i,j],
(sum[j]-sum[i-1])/(j-i+1)>a <=> sum[j]-sum[i-1]>a*(j-i+1)
<=> sum[j]-sum[i-1]-a*(j-i+1)>0
<=> s[j]-a+s[j-1]-a+...+s[i]-a>0
令s'[i]=s[i]-a,则sum'[i]=sum[i]-i*a
那么(sum[j]-sum[i-1])/(j-i+1)>a <=> sum'[j]-sum'[i]>0
<=> sum'[i]<sum'[j] i<j
意思就是把一段区间的平均值大于a转化成先把每个数减去a,然后这个区间的和大于0
这样,我们只要求有多少对sum'[i]和sum'[j]满足上述条件就行了,跟求逆序对数 差不多
注意答案要用__int64保存
代码: