洛谷 P3367 【模板】并查集 + P1455 搭配购买

1.P3367【模板】并查集

题目描述

如题，现在有一个并查集，你需要完成合并和查询操作。

输入格式

第一行包含两个整数 $N,M$ ,表示共有 $N$ 个元素和 $M$ 个操作。

接下来 $M$ 行，每行包含三个整数 $Z_i,X_i,Y_i$ 。

当 $Z_i=1$ 时，将 $X_i$ 与 $Y_i$ 所在的集合合并。

当 $Z_i=2$ 时，输出 $X_i$ 与 $Y_i$ 是否在同一集合内，是的输出
Y ；否则输出 N 。

输出格式

对于每一个 $Z_i=2$ 的操作，都有一行输出，每行包含一个大写字母，为 Y 或者 N 。

样例 #1

样例输入 #1

4 7
2 1 2
1 1 2
2 1 2
1 3 4
2 1 4
1 2 3
2 1 4

样例输出 #1

N
Y
N
Y

提示

对于 $30\%$ 的数据，$N\le 10$，$M\le 20$。

对于 $70\%$ 的数据，$N\le 100$，$M\le 10^3$。

对于 $100\%$ 的数据，$1\le N\le 10^4$，$1\le M\le 2\times 10^5$，$1\le X_i,Y_i\le N$， Z i ∈ { 1 , 2 } Z_i \in \{ 1, 2 \} Zi​∈{1,2}。

AC代码

#include<bits/stdc++.h>
//#define int long long
#define LF(x) fixed << setprecision(x)
using namespace std;
const int t=1e6+100;
int a[t];
int n,m;
int find(int g)//利用递归查找g的“老大”
{
    if(a[g]!=g) a[g]=find(a[g]);
    return a[g];
}
signed main()
{
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin>>n>>m;
    for(int i=1;i<=n;i++) a[i]=i;//初始化，使每个数的“老大”都为自己
    for(int i=0; i<m; i++)
    {
        int z,x,y;
        cin>>z>>x>>y;
        if(z==1)//合并集合（找到x,y的“老大”）
        {
        	//如果x,y的老大不相同，就合并使x的老大变成y的老大
            if(find(x)!=find(y))
            a[find(x)]=find(y);
        }
        else{
            if(find(x)==find(y))
            cout<<"Y"<<endl;
            else
            cout<<"N"<<endl;
        }
    }
    return 0;
}

2.P1455 搭配购买(并查集+背包问题)

题目描述

明天就是母亲节了，电脑组的小朋友们在忙碌的课业之余挖空心思想着该送什么礼物来表达自己的心意呢？听说在某个网站上有卖云朵的，小朋友们决定一同前往去看看这种神奇的商品，这个店里有 $n$ 朵云，云朵已经被老板编号为 $1,2,3,...,n$，并且每朵云都有一个价值，但是商店的老板是个很奇怪的人，他会告诉你一些云朵要搭配起来买才卖，也就是说买一朵云则与这朵云有搭配的云都要买，电脑组的你觉得这礼物实在是太新奇了，但是你的钱是有限的，所以你肯定是想用现有的钱买到尽量多价值的云。

输入格式

第一行输入三个整数，$n,m,w$，表示有 $n$ 朵云，$m$ 个搭配和你现有的钱的数目。

第二行至 $n+1$ 行，每行有两个整数， $c_i,d_i$，表示第 $i$ 朵云的价钱和价值。

第 $n+2$ 至 $n+1+m$ 行 ，每行有两个整数 $u_i,v_i$。表示买第 $u_i$ 朵云就必须买第 $v_i$ 朵云,同理，如果买第 $v_i$ 朵就必须买第 $u_i$ 朵。

输出格式

一行，表示可以获得的最大价值。

样例 #1

样例输入 #1

5 3 10
3 10
3 10
3 10
5 100
10 1
1 3
3 2
4 2

样例输出 #1

1

提示

• 对于 $30\%$ 的数据，满足 $1\le n\le 100$；
• 对于 $50\%$ 的数据，满足 $1\le n,w\le 10^3$，$1\le m\le 100$；
• 对于 $100\%$ 的数据，满足 $1\le n,w\le 10^4$，$0\le m\le 5\times 10^3$。

AC代码

因为云朵要捆绑购买，即合并云朵（并查集），又要求在有限的money下所获得的价值最大，所以为01背包问题，只需再dp一下就好

#include<bits/stdc++.h>
// #define int long long
#define LF(x) fixed << setprecision(x)
using namespace std;
const int t = 1e4 + 100;
int v[t], k[t], p[t], dp[t];
int n, m, w;
vector<pair<int, int>> bag;//使用动态数组存入每次购买所需的钱与获得价值
int find(int cur)//并查集模版
{
    if (p[cur] != cur)
        p[cur] = find(p[cur]);
    return p[cur];
}
signed main()
{
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin >> n >> m >> w;
    for (int i = 1; i <= n; i++)
    {
        cin >> v[i] >> k[i];
    }
    for (int i = 1; i <= n; i++)//初始化，使每个数的“老大”都为自己
        p[i] = i;
    for (int i = 1; i <= m; i++)
    {
        int x, y;
        cin >> x >> y;
        if (find(x) != find(y))
        {
            p[find(x)] = find(y);
        }
    }
    //重新遍历使“老大”（即捆绑云朵）的所需钱与获得价值相加
    for (int i = 1; i <= n; i++)
    {
        if (p[i] != i)
        {
            v[find(i)] += v[i];
            k[find(i)] += k[i];
        }
    }
    //将“老大”的数据存入背包
    for (int i = 1; i <= n; i++)
    {
        if (p[i] == i)
            bag.push_back({v[i], k[i]});
    }
    //01背包模版 这一点如果不懂可以去看一下[背包九讲](http://t.csdnimg.cn/PdS8N)
    for (auto i : bag)
        for (int j = w; j >= i.first; j--)
            dp[j] = max(dp[j], dp[j - i.first] + i.second);
    cout << dp[w] << endl;
    return 0;
}

蒟蒻刚开始写题解，如有不对欢迎大佬指出
感谢
完结撒花