CSP202012-4食材运输

题目描述

寻找规律

• 对于一种食材的特定的检查点将食材运送到所有酒店的运行时间为：到所有需求食材酒店边权的两倍减最长路径长度；为什么？
  • 除了最远路径上的边外，其余边均回退一次，前进一次。（多画几个找规律）

M=K的做法

• 用栈st维护dfs过程种遇到的边，到达需求结点时加入栈中没有累加过的所有边。
• 树无环，dfs判断是否不能回退只需不回退到来的结点。

#include<iostream>
#include<algorithm>
#include<map>
#include<stdio.h>
#include<string.h>
#include<unordered_map>
#include<vector>
#include<queue>
#include<set>
using namespace std;

struct edge{
    int u;
    int v;
    int w;
};

void dfs(int from,int sr,bool vis[],bool req[],struct edge ed[],vector<int> adj[],long long &sum,long long & max_now,vector<int> st,long long total)
{
    if(req[sr])
    {
        for(int i = st.size() - 1;i>=0 && !vis[st[i]];i--) {vis[st[i]] = true; sum+=(long long)ed[st[i]].w;}
        max_now = max(max_now,total);
    }
    for(int e : adj[sr])  //从源节点开始的每一条边
    {
        int to  = ed[e].u == sr ? ed[e].v : ed[e].u;
        if(to!=from)
        {
            st.push_back(e);
            dfs(sr,to,vis,req,ed,adj,sum,max_now,st,total + (long long)ed[e].w);
            st.pop_back();
        }
    }
}

int main()
{
    int N,M,K;
    cin >> N >> M >> K;
    bool req[K][N];         //
    long long cost[K][N];   //第k种食材在第i个酒店开始运输所需要的最小开销
    vector<int> adj[N];            //每个结点和哪些边相关联
    struct edge ed[N];
    for(int i = 0;i<N;i++)
    {
        for(int j= 0;j<K;j++)
        {
            cost[j][i] = 0;
            cin >> req[j][i];  //第j种食材第i个酒店是否需要   
        }
    }
    for(int i = 0;i<N-1;i++)
    {
        cin >> ed[i].u>> ed[i].v>> ed[i].w;
        ed[i].u--; ed[i].v--;
        adj[ed[i].u].push_back(i); adj[ed[i].v].push_back(i);
    }
    bool vis[N - 1];   //边是否被相加过
    vector<int> st;    //保存经过的边
    for(int i = 0;i<K;i++)
    {
        for(int j = 0;j<N;j++)
        {
            memset(vis,0,N-1);
            long long sum = 0;
            long long max_now =  -9e18;
            dfs(-1,j,vis,req[i],ed,adj,sum,max_now,st,0);
            cost[i][j] = sum * 2 - max_now;
        }
    }
    long long ans = -1;
    //对于每种食材选最小的
    for(int i = 0;i<K;i++)
    {
        long long minti = 9e18;
        for(int j = 0;j<N;j++)
            minti = min(minti,cost[i][j]);
        ans = max(ans,minti);
    }
    cout << ans << endl;
    //system("pause");
    return 0;
}

$M\le K$的做法