Codeforces Round #422 (Div. 2) C. Hacker, pack your bags! 排序, 贪心, 暴力(居然可以n方暴力)

题目链接: Hacker, pack your bags!

题目大意

有x天的假期, 有n张旅行票, 每张票有起始时间l, 结束时间r, 花费cost, 想把假期分成两部分出去旅游, 两部分时间不能重合(ri < lj || rj < li), 问最小花费是多少, 如果不能两部分, 输出-1

思路

1. 排序, 贪心

CF官方解法, 效率$O(nlog_2^n)$
设置一个结构体, struct P{int p, len, cost, type};
将每张票(l, r, cost) 表示成两个结构体, P(l, r-l+1, cost, -1), P(r, r-l+1, cost, 1);
设置一个数组best[i], 表示时间长度为i的最便宜的票, 一开始全为INF, 之后边用边更新
将结构体数字排序, 首先按p排序, p相同按type排序, 这样保证了是按时间顺序且同样时间type为-1的在type为1的前面
遍历整个结构体数组
type为1, 则用p[i].cost更新best[p[i].len]
type为-1, 则用p[i].cost+best[x-p[i].len]更新ans
因为数组按照p然后type排序, 保证了更新ans是用到的best[], 里面存储的都是根据时间段在之前的票得出的, 保证时间不会重叠

代码

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const int MAXN = 2E5 + 100, INF = 2e9+10;
int n, x, l, r, c;
int best[MAXN];
struct P
{
    int p, len, cost, type;
    bool operator<(const P &x)
    {
        if(p == x.p) return type < x.type;
        return p < x.p;
    }
}p[MAXN*2];


int main()
{
    scanf("%d%d", &n, &x);
    int cnt = 0;
    for(int i=0; i<n; ++i)
    {
        scanf("%d%d%d", &l, &r, &c);
        p[cnt++] = P{l, r-l+1, c, -1};
        p[cnt++] = P{r, r-l+1, c, 1};
    }
    ll ans = INF;
    sort(p, p+cnt);
    fill(best, best+x, INF);

    for(int i=0; i<cnt; ++i)
    {
        if(p[i].type == -1)
        {
            if(p[i].len < x)
            {
                ans = min(ans, (ll)p[i].cost+(ll)best[x-p[i].len]);
            }
        }
        else best[p[i].len] = min(best[p[i].len], p[i].cost);
    }

    if(ans >= INF) ans = -1;
    cout << ans <<endl;

    return 0;
}

2. 暴力

CF上看到好多人都是暴力过去的…
复杂$O(n^2)$加了一点微小的优化, 结果在CF上时间和第一种做法一样
开一个vector<pair<int,int>> vp , 长度为i的票存储在vp[i]里面, first表示开始时间l, second表示cost
然后循环1-x, 在两个循环遍历vp[i], vp[x-i], 检查时间是否重叠, 更新答案
把每个vp[i]根据l排序一下, 可以优化一点

代码

#include <bits/stdc++.h>

using namespace std;

const int MAXN = 3E5 + 100, INF = 0X7FFFFFFF;
int n, x, l, r, c;
vector<pair<int, int>> vp[MAXN];


int main()
{
    scanf("%d%d", &n, &x);
    for(int i=0; i<n; ++i)
    {
        scanf("%d%d%d", &l, &r, &c);
        vp[r-l+1].push_back({l, c});
    }
    for(int i=0; i<=x; ++i) sort(vp[i].begin(), vp[i].end());

    int ans = INF;

    for(int d=1; d<x; ++d)
    {
        int mn = INF;
        auto & u=vp[d], &v = vp[x-d];

        for(int i=0, j=0; i<v.size(); ++i)
        {
            while(j<u.size() && u[j].first+d-1<v[i].first)
            {
                if(mn > u[j].second) mn = u[j].second;
                ++j;
            }
            if(mn != INF)
            {
                int s = mn + v[i].second;
                if(ans > s) ans = s;
            }
        }

    }
    if(ans == INF) cout << -1 << endl;
    else cout << ans << endl;

    return 0;
}