CF1618G Trader Problem

原题链接 ← 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