CF1618G Trader Problem

2022-02-09 12:32:14 阅读：176 来源： 互联网

标签：rnk int Trader ++ vector CF1618G 物品 Problem first

原题链接 ← Click it
这题难产了快一个月终于决定要攻克下来
我好菜啊

题目大意：现在你手里有\(n\)件物品，它们的价值分别是\(a_1,a_2,...,a_n\)，你现在可以和一个商人交换物品，他有\(m\)件物品，价值分别为\(b_1,b_2,...,b_m\)，价值为\(w\)的物品，最多可以与价值为\(w + k\)的物品交换，现在你有\(q\)次询问，每次询问一个\(k\)，问你可以交换到的总物品的价值最大值。

解题思路：考虑并查集维护每个物品能交换到的所以物品的集合，在每个集合中从大到小依次取出该集合中属于初始物品的个数个物品进行求和。优化1：对于任意的一个\(k\)，你对每一件物品询问它能交换到的价值最大值都需要从原始的价值出发，因为你不能保证下一次一定能从已经交换后的价值发出(时间复杂度\(O(qn^2)\))，所以考虑\(q\)次离线询问，对\(k_i\)进行排序后，每次询问都和合理利用上次询问的信息。优化2：因为每次询问合并的物品只需要满足和前一个相邻物品的差值\(> k_{i - 1}\ \&\&\ <= k_{i}\)，所以可以对每个相邻物品的价值差进行排序，用双指针将每次询问需要进行合并的序号都加入队列。优化3：每次合并之后都需要对重新计算每个集合对答案的贡献，如果是减去合并前集合的贡献再加上合并后集合的贡献，时间复杂度达到\(O(n)\)，但是如果预先求前缀和，每次重新计算贡献的时间复杂度就达到了\(O(1)\)。

参考代码：

typedef long long ll;
typedef pair<int, int> PII;
const int INF = 0x3f3f3f3f, N = 1e6 + 10;
struct DSU {
    int N;
    vector<int> pr, rnk;
    DSU(int n) : N(n) {
        pr.resize(N);
        rnk.resize(N);
        for(int i = 0; i < N; i ++) {
            pr[i] = i;
            rnk[i] = 1;
        }
    }
    int root(int x) {
        return pr[x] == x ? x : pr[x] = root(pr[x]);
    }
    bool same(int a, int b) {
        return root(a) == root(b);
    }
    bool unite(int a, int b) {
        int A = root(a), B = root(b);
        if(A == B) return 0;
        if(rnk[A] == rnk[B]) rnk[A] ++;
        else if(rnk[A] > rnk[B]) swap(rnk[A], rnk[B]);
        pr[A] = B;
        return 1;
    }
};
void solve() {
    int n, m, q; ll sum = 0;
    cin >> n >> m >> q;
    vector<PII> a(n + m);
    vector<ll> b(n + m + 1);
    vector<int> siz(n + m);//每个集合中属于初始物品的个数
    vector<PII> derta;
    for(int i = 0; i < n + m; i ++) {
        cin >> a[i].first;
        //标记初始物品，并求初始价值和
        if(i < n) {
            a[i].second = 1;
            sum += a[i].first;
        }
    }
    sort(a.begin(), a.end());
    for(int i = 0; i < n + m; i ++) {
        if(i) derta.push_back({a[i].first - a[i - 1].first, i});
        siz[i] += a[i].second;
    }
    for(int i = 1;i < n + m + 1; i ++) {
        b[i] = b[i - 1] + a[i - 1].first;
    }
    vector<PII> query(q);
    vector<vector<int>> qa(q);
    vector<ll> res(q);
    for(int i = 0; i < q; i ++) {
        cin >> query[i].first;
        query[i].second = i;
    }
    sort(query.begin(), query.end());
    sort(derta.begin(), derta.end());
    for(int i = 0, j = 0; i < q && j < n + m - 1; i ++) {
        while(j < n + m - 1 && derta[j].first <= query[i].first) {
            qa[i].push_back(derta[j].second);
            j ++;
        }
    }
    DSU tree(n + m);
    for(int i = 0; i < q; i ++) {
        int k = query[i].first, lx = query[i].second;
        for(auto x : qa[i]) {
            if(tree.unite(x - 1, x)) {
                sum -= b[x] - b[x - siz[x - 1]];
                int tmp = tree.root(x);
                sum += b[tmp + 1 - siz[tmp]] - b[tmp + 1 - siz[tmp] - siz[x - 1]];
                siz[tmp] += siz[x - 1];
            }
        }
        res[lx] = sum;
    }
    for(int i = 0; i < q; i ++) {
        cout << res[i] << '\n';
    }
}

tag：并查集、双指针、前缀和、贪心、离线询问

标签：rnk,int,Trader,++,vector,CF1618G,物品,Problem,first
来源： https://www.cnblogs.com/Muly/p/15874511.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

CF1618G Trader Problem