SOJ 1046 Plane Spotting

题目大意：输入整数t，表示有t组测试样例。每组测试样例首先输入三个整数，分别是n(1 ≤ n ≤300)表示有n个观测飞机的时刻，l(1 ≤ l ≤ 100)表示输出结果中的的最高l组，m(1 ≤ m ≤ 300)表示时刻区间的最少包含区间数。接着输入n个整数，分别代表每个时刻可观测的飞机数，计作p_i(1 ≤ i ≤ n)。求解满足最少区间要求的区间中平均时刻能观测飞机数最多的l组区间。

其他条件：

区间p_i比p_j好的条件是：

1.p_i区间平均时刻飞机数多余p_j区间

2.若平均时刻飞机数相同，则需要p_i区间大于p_j区间

3.若平均时刻飞机数相同写，区间大小相同，则需要p_i比p_j 早完成。

输出格式：

　　先输出"Result for run #:"，其中#表示第几组测试样例，从1开始。

　　再输出最高l组区间，"{from}-{to}"，{from}表示区间开始，{to}表示区间结束。

解题思路：首先选出所有满足最少时刻数要求的区间，然后对每个区间进行排序比较。最后最高的l组结果，如果满足要求的区间不足l组，则按顺序输出全部顺序。

　　　　　考虑到求区间和，想起了前缀和，这是一个小小的优化。

代码如下：

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

const int maxn = 305;
int arr[maxn];
int sum[maxn];

struct Interval {
    int from;
    int to;
    Interval(int f = 0, int t = 0) {
        from = f;
        to = t;
    }
};

bool cmpFunc(const Interval & i1, const Interval & i2) {
    bool swap;
    double s1 = (sum[i1.to] - sum[i1.from - 1] + 0.0) / (i1.to - i1.from + 1);
    double s2 = (sum[i2.to] - sum[i2.from - 1] + 0.0) / (i2.to - i2.from + 1);
    if (s1 > s2) {
        swap = false;
    } else if (s1 < s2) {
        swap = true;
    } else {
        int l1 = i1.to - i1.from + 1;
        int l2 = i2.to - i2.from + 1;
        if (l1 > l2) {
            swap = false;
        } else if (l1 < l2) {
            swap = true;
        } else {
            if (i1.to > i2.to) {
                swap = true;
            } else if (i1.to < i2.to) {
                swap = false;
            } else {
                cout << "error this time" << endl;
            }
        }
    }
    return !swap;
}

int main() {
    int t, n, l, m;
    
    cin >> t;
    for (int tt = 1; tt <= t; tt++) {
        vector<Interval> vt;
        cin >> n >> l >> m;

        // 获取每时刻的飞机数
        arr[0] = sum[0] = 0;
        for (int i = 1; i <= n; i++) {
            cin >> arr[i];
            sum[i] = 0;
        }

        // 计算前缀和
        for (int i = 1; i <= n; i++) {
            sum[i] = sum[i - 1] + arr[i];
        }

        // 获得所有可行的区间
        for (int i = 1; i + m - 1 <= n; i++) {
            for (int j = i + m - 1; j <= n; j++) {
                vt.push_back(Interval(i, j));
            }
        }

        // 对区间信息进行排序
        sort(vt.begin(), vt.end(), cmpFunc);

        // 输出结果
        cout << "Result for run " << tt << ":" << endl;
        int c = l > vt.size() ? vt.size() : l;
        for (int i = 0; i < c; i++) {
            cout << vt[i].from << "-" << vt[i].to << endl;
        }

    }

    return 0;
}