Acwing算法基础课学习笔记(十七)--贪心之区间问题&&Huffman树

区间选点

1. 将每个区间按照右端点从小到大进行排序

2. 从前往后枚举区间，end值初始化为无穷小

   • 如果本次区间不能覆盖掉上次区间的右端点， ed < range[i].l

   • 说明需要选择一个新的点， res ++ ; ed = range[i].r;

   • 如果本次区间可以覆盖掉上次区间的右端点，则进行下一轮循环

证明

• 证明ans <= cnt ：cnt 是一种可行方案， ans是可行方案的最优解，也就是最小值。
• 证明ans >= cnt ： cnt可行方案是一个区间集合，区间从小到大排序，两两之间不相交。所以覆盖每一个区间至少需要cnt个点。

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010;

int n;
//结构体定义的是每一个区间，l和r代表左右端点
struct Range
{
    int l, r;
    //由于要排序，所以需要重载小于号'<'
    bool operator< (const Range &W)const
    {
        return r < W.r;
    }
}range[N];

int main()
{
    scanf("%d", &n);
    for (int i = 0; i < n; i ++ )
    {
        int l, r;
        scanf("%d%d", &l, &r);
        range[i] = {l, r};
    }

    sort(range, range + n);//排序
    int res = 0, ed = -2e9;//最开始的时候一个点都没选，所以初始化成负无穷
    
    for (int i = 0; i < n; i ++ )//枚举所有区间
        if (range[i].l > ed)//当前区间的左端点严格大于ed
        {
            res++;//需要选择新的点
            ed = range[i].r;//将当前点的下标更新成右端点
        }

    printf("%d\n", res);

    return 0;
}

最大不相交区间数量
做法其实和上一个问题是完全一样的。

1. 将每个区间按右端点从小到大排序
2. 从前往后依次枚举每个区间，如果当前区间中已经包含点，则直接pass，否则，选择当前区间的右端点

代码一毛一样hh：

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010;

int n;
//结构体定义的是每一个区间，l和r代表左右端点
struct Range
{
    int l, r;
    //由于要排序，所以需要重载小于号'<'
    bool operator< (const Range &W)const
    {
        return r < W.r;
    }
}range[N];

int main()
{
    scanf("%d", &n);
    for (int i = 0; i < n; i ++ )
    {
        int l, r;
        scanf("%d%d", &l, &r);
        range[i] = {l, r};
    }

    sort(range, range + n);//排序
    int res = 0, ed = -2e9;//最开始的时候一个点都没选，所以初始化成负无穷
    
    for (int i = 0; i < n; i ++ )//枚举所有区间
        if (range[i].l > ed)//当前区间的左端点严格大于ed
        {
            res++;//需要选择新的点
            ed = range[i].r;//将当前点的下标更新成右端点
        }

    printf("%d\n", res);

    return 0;
}

区间分组

#include <iostream>
#include <algorithm>
#include <queue>
#include <cstdio>

using namespace std;

const int N = 100010;

int n;
struct Range
{
    int l, r;
    bool operator< (const Range &W)const
    {
        return l < W.l;
    }
}range[N];

int main()
{
    scanf("%d", &n);
    for (int i = 0; i < n; i ++ )
    {
        int l, r;
        scanf("%d%d", &l, &r);
        range[i] = {l, r};
    }

    sort(range, range + n);

    priority_queue<int, vector<int>, greater<int>> heap;
    for (int i = 0; i < n; i ++ )
    {

        if (heap.empty() || heap.top() >= range[i].l){
            heap.push(range[i].r);
        }
        else {
            heap.pop();
            heap.push(range[i].r);
        }
    }

    printf("%d\n", heap.size());

    return 0;
}

区间覆盖

1. 将所有区间按照左端点从小到大进行排序
2. 从前往后枚举每个区间，在所有能覆盖start的区间中，选择右端点的最大区间，然后将start更新成右端点的最大值;这一步用到了贪心决策

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010;

int n;
struct Range
{
    int l, r;
    bool operator< (const Range &W)const
    {
        return l < W.l;
    }
}range[N];

int main()
{
    int st, ed;
    scanf("%d%d", &st, &ed);
    scanf("%d", &n);
    for (int i = 0; i < n; i ++ )
    {
        int l, r;
        scanf("%d%d", &l, &r);
        range[i] = {l, r};
    }

    sort(range, range + n);

    int res = 0;
    bool success = false;
    for (int i = 0; i < n; i ++ )
    {
        int j = i, r = -2e9;
        while (j < n && range[j].l <= st)
        {
            r = max(r, range[j].r);
            j ++ ;
        }

        if (r < st)
        {
            res = -1;
            break;
        }

        res ++ ;
        if (r >= ed)
        {
            success = true;
            break;
        }

        st = r;
        i = j - 1; 
    }

    if (!success) res = -1;
    printf("%d\n", res);

    return 0;
}

合并果子

#include <iostream>
#include <queue>
#include <algorithm>

using namespace std;

int main()
{
    int n;
    cin >> n;
    priority_queue <int, vector<int>, greater<int>> heap;
    while (n--)
    {
        int x;
        cin >> x;
        heap.push(x);
    }
    
    int res;
    while (heap.size() > 1)
    {
        int a = heap.top(); heap.pop();
        int b = heap.top(); heap.pop();
        res += a + b;
        heap.push(a + b);
    }
    
    cout << res << endl;
    return 0;
}