合成反应

链接：https://www.nowcoder.net/acm/contest/78/I
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

有机合成是指从较简单的化合物或单质经化学反应合成有机物的过程。

有时也包括从复杂原料降解为较简单化合物的过程。

由于有机化合物的各种特点，尤其是碳与碳之间以共价键相连，有机合成比较困难，常常要用加热、光照、加催化剂、加有机溶剂甚至加压等反应条件。

但是前人为有机合成提供了许多宝贵的经验。

现在已知有K总物质和N个前人已经总结出的合成反应方程式

小星想知道在现有M种物质的情况下 能否合成某些物质。

输入描述:

第一行输入四个整数 K,N,M,Q（K,N,M,Q<=1e5）
K表示一共K总物质
接下来N行 每行三个数字a b c（任意两个数可能相等）
表示a和b反应可以生成c 反应是可逆的
即可以通过c可以分解出a和b
接下来一行行然后输入m个数，表示m种原料（每一种原料都可以认为有无限多）
接下来Q个行Q个询问
对于每个询问
输出一个数字 x 判断是否可以通过一些反应得到第 x

输出描述:

可以得到Yes否则No

示例1

输入

10 3 4 10
1 2 3
4 5 6
2 5 7
3 4 5 8
1
2
3
4
5
6
7
8
9
10

输出

Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
No
No

说明

一共10总物质有第3，4，5，8 四种原料
查询每一种是否可以通过反应得到
首先通过3可以分解得到1 2
然后4 5合成6
2 5合成7
于是除了9 10都可以得到

我怕超时就用了向量

#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <vector>
#define inf 0x3f3f3f3f
using namespace std;
//const double pi = acos(-1), e = exp(1);
vector<int>a[100005];
bool v[100005];
bool ff[100005];
void broad(int x)//如果x可以，那么x分解的都可以
{
    int i;
    ff[x] = 1;
    for (i = 0; i<a[x].size(); i++)
    {
        int t1 = a[x].at(i);
        if (v[t1] == 0)
        {
            v[t1] = 1;
            if (a[t1].size() >= 2 && ff[t1] == 0)
            {
                broad(t1);
            }
        }
    }
}
bool bb(int x)//如果x不可以，那么查看合成它的行不行
{
    int i;
    for (i = 0; i < a[x].size(); i=i+2)
    {
        int t1 = a[x].at(i);
        int t2 = a[x].at(i + 1);
        if (v[t1] == 1 && v[t2] == 1)
        {
            v[x] = 1;
            if(a[x].size()>2) broad(x);
            return 1;
        }
    }
    return 0;
}
int main()
{
    int k, n, m, q;
    int mm = -inf;
    int mi = inf;
    cin >> k >> n >> m >> q;
    memset(v, 0, sizeof(v));
    memset(ff, 0, sizeof(ff));
    int i;
    int x, y, z;
    for (i = 1; i <= 10005; i++) a[i].clear();
    for (i = 1; i <= n; i++)
    {
        cin >> x >> y >> z;
        if (x > mm) mm = x;
        if (x < mi) mi = x;
        if (y > mm) mm = y;
        if (y < mi) mi = y;
        if (z > mm) mm = z;
        if (z < mi) mi = z;

        a[z].push_back(x);
        a[z].push_back(y);
    }
    int p;
    for (p = 1; p <= m; p++)
    {
        cin >> x;
        v[x] = 1;
        if (a[x].size()>=2&&ff[x] == 0)
        {
            broad(x);//如果x可以，那么x分解的都可以
        }
    }
    bool up = 1;
    while (up)//查看合成它的
    {
        up = 0;
        for (i = mi; i <= mm; i++)
        {
            if (v[i]==0&&a[i].size() >= 2) 
            {
                bool f ;
                f = bb(i);
                if (f) up = 1;
            }
        }
    }
    for (i = 1; i <= q; i++)
    {
        cin >> x;
        if (v[x] == 1) cout << "Yes" << endl;
        else cout << "No" << endl;
    }
    return 0;
}

别人的代码更简单，不用向量也可以，直接暴力

#include <iostream>
#include <cstring>
using namespace std;
 
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
 
    int k, n, m, q;
    cin >> k >> n >> m >> q;
    bool book[k + 1];
    memset(book, false, sizeof(bool) * (k + 1));
 
    int first1[n], first2[n], second[n];
    for (int i = 0; i < n; ++i)
        cin >> first1[i] >> first2[i] >> second[i];
 
    for (int i = 0; i < m; ++i)
    {
        int t;
        cin >> t;
        book[t] = true;
    }
 
    bool flag = true;
    while (flag)
    {
        flag = false;
        for (int i = 0; i < n; ++i)
        {
            if (book[first1[i]] && book[first2[i]] && !book[second[i]])
            {
                book[second[i]] = true;
                flag = true;
            }
            if (book[second[i]] && (!book[first1[i]] || !book[first2[i]]))
            {
                book[first1[i]] = true;
                book[first2[i]] = true;
                flag = true;
            }
        }
    }
 
    for (int i = 0; i < q; ++i)
    {
        int t;
        cin >> t;
        cout << (book[t] ? "Yes" : "No") << endl;
    }
 
    return 0;
}