蓝桥杯真题——作物杂交

题目描述

解析

通过不断的将已有的种子杂交得到新种子，直到得到目标种子是比较麻烦的，因此我们考虑从目标种子开始倒推到已有的种子，好比走迷宫从出口倒着走回到入口。

对输入的进来的杂交方案，使用vector<vector<pair<int,int>>>记录，以第一维下标表示杂交结果，杂交的需要的种子放进pair里，以cost数组记录得到每个种子所需要的最短时间，初始有的种子对应时间为0，而其余初始化为最大值。

从得到目标种子的杂交方案开始往前推，如果需要的种子的最短时间还没确定，则递归计算，这样不断地得到中间过程中每个种子所花费的最短时间，递归结束即得到当前种子所花费的最短时间。

完整代码

#include<iostream>
#include<vector>

using namespace std;

void dfs(vector<int>& cost, vector<vector<pair<int, int> > >& hybrid, vector<int>& t, int target) {
    for (int i = 0; i < hybrid[target].size(); i++) {
        int a = hybrid[target][i].first;
        int b = hybrid[target][i].second;
        if (cost[a] == INT_MAX) {
            dfs(cost, hybrid, t, a);
        }
        if (cost[b] == INT_MAX) {
            dfs(cost, hybrid, t, b);
        }
        cost[target] = min(cost[target], max(t[a], t[b]) + max(cost[a], cost[b]));
    }
}
int main()
{
    int N, M, K, T;
    cin >> N >> M >> K >> T;
    vector<int> t(N + 1); //种植时间
    for (int i = 1; i <= N; ++i) {
        cin >> t[i];
    }
    vector<int> cost(N + 1, INT_MAX); //花费总时间
    int tmp;
    for (int i = 0; i < M; ++i) {
        cin >> tmp;
        cost[tmp] = 0;
    }
    int a, b, c;
    vector<vector<pair<int, int> > > hybrid(N + 1); //杂交方案
    for (int i = 0; i < K; ++i) {
        cin >> a >> b >> c;
        pair<int, int> pa = make_pair(a, b);
        hybrid[c].push_back(pa);
    }
    dfs(cost, hybrid, t, T);
    cout << cost[T] << endl;
    return 0;
}