第十一届蓝桥杯c++研究生组--H作物杂交

题目连接

代码如下：

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=1e5+10;
vector <pair<int,int>> f[maxn];
int t[maxn];//时间
int k;//种子类型
int n,m,K,T;
//int f[maxn][maxn];
int vis[maxn];
int d[maxn];
void dfs(int u)
{
    for(int i=0;i<f[u].size();i++)
    {
        int a=f[u][i].first;
        int b=f[u][i].second;
        if(vis[a]==0)//没有a
           dfs(a);//先去培养得到a
        if(vis[b]==0)//没有b
           dfs(b);//先去培养b
        if(vis[a]&&vis[b])//a和b都有了，可以培养u了，计算最短时间
        {
            vis[u]=1;
            d[u]=min(d[u],max(t[a],t[b])+max(d[a],d[b]));
        }
    }
}
int main()
{
    cin>>n>>m>>K>>T;
    for(int i=1;i<=n;i++)
    {
        cin>>t[i];
        d[i]=99999999;
    }
    for(int i=1;i<=m;i++)
    {
        cin>>k;//已经拥有的m个种子类型
        vis[k]=1;
        d[k]=0;
    }
    for(int i=1;i<=K;i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        f[c].push_back({a,b});
    }
    dfs(T);//T是想要得到的种子
    cout<<d[T]<<endl;
    return 0;
}