MT2057 门票

 

思路：

此题是求有多少个区间的平均值>=t， 那么可以把每个值-t。如果新的数列的某个区间的和>=0，那么说明这个区间满足条件。

令新数列的前缀和为b[i]，所以求[i, j]区间是否满足条件，即求b[j]-b[i-1]是否>=0，即b[j]>=b[i-1]。

因为j>i>i-1，所以这里即求“伪逆序对”的数量。

扩展知识：

逆序对：i>j a[i]<a[j]      伪逆序对/非逆序对：i>j a[i]>a[j]

方法：归并排序

代码：

1.8/10代码：错误原因：超时

#include <bits/stdc++.h>
using namespace std;
const long long int N = 1e6 + 10;
long long int p = 1e9 + 7;
long long int n, t;
long long int a[N];
long long int b[N];
int main()
{
    cin >> n >> t;
    for (long long int i = 1; i <= n; i++)
    {
        cin >> a[i];
        a[i] -= t;
    }
    for (long long int i = 1; i <= n; i++)
    {
        b[i] = b[i - 1] + a[i];
    }
    long long int ans = 0;
    for (long long int i = 1; i <= n; i++)
    {
        for (long long int j = 1; j <= i; j++)
        {
            if (b[i] - b[j - 1] >= 0)
            {
                ans++;
            }
        }
    }
    cout << ans % p;
}

2.10/10代码：升序排列求逆序对，再用总的-逆序对即为非逆序对个数

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 1e6 + 10;
int p = 1e9 + 7;
ll n, t;
ll a[N], sum[N], q[N];
ll ans = 0;
void merge_sort(int l, int r, ll a[])
{
    if (l >= r)
        return;
    int mid = (l + r) >> 1;

    merge_sort(l, mid, a);
    merge_sort(mid + 1, r, a);

    int i = l, j = mid + 1, k = 0;
    while (i <= mid && j <= r)
    {
        if (a[i] > a[j])
        {
            q[k++] = a[j++];
            ans += mid - i + 1; // 升序排列，求逆序数
            ans %= p;
        }
        else
        {
            q[k++] = a[i++];
        }
    }
    while (i <= mid)
        q[k++] = a[i++];
    while (j <= r)
        q[k++] = a[j++];
    for (i = l, j = 0; i <= r; i++, j++)
    {
        a[i] = q[j];
    }
}

int main()
{
    cin >> n >> t;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        a[i] -= t;
        sum[i] = sum[i - 1] + a[i];
    }
    merge_sort(0, n, sum);
    cout << (n * (n + 1) / 2 - ans) % p;
    return 0;
}

3.10/10代码，直接降序求非逆序对个数

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 1e6 + 10;
int p = 1e9 + 7;
ll n, t;
ll a[N], sum[N], q[N];
ll ans = 0;
void merge_sort(int l, int r, ll a[])
{
    if (l >= r)
        return;
    int mid = (l + r) >> 1;

    merge_sort(l, mid, a);
    merge_sort(mid + 1, r, a);

    int i = l, j = mid + 1, k = 0;
    while (i <= mid && j <= r)
    {
        if (a[i] <= a[j])
        {
            q[k++] = a[j++];
            ans += mid - i + 1; // 降序排列，求非逆序数
            ans %= p;
        }
        else
        {
            q[k++] = a[i++];
        }
    }
    while (i <= mid)
        q[k++] = a[i++];
    while (j <= r)
        q[k++] = a[j++];
    for (i = l, j = 0; i <= r; i++, j++)
    {
        a[i] = q[j];
    }
}

int main()
{
    cin >> n >> t;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        a[i] -= t;
        sum[i] = sum[i - 1] + a[i];
    }
    merge_sort(0, n, sum);
    cout << ans % p;
    return 0;
}