稀有矿井

D: 稀有矿井

时间限制: 1 Sec  内存限制: 128 MB

题目描述

XYZ公司已在沿太平洋东海岸位于不同地区的几个岛屿发现了5种稀有的矿藏。该公司认为，这将是一个获利最好的机会。然而，由于金融危机，该公司并没有足够的人手和金钱在所有岛屿上建立矿田。因此，公司委员会选择了一些有较高的矿石储量岛屿，并派出一名调查员对这些岛屿制作了岛上的矿石分布调查。调查结果显示，每个岛上有许多连接在一起的村庄。由于耗费时间，调查员并没有记录的地图中的所有路径。只是记下了一条到达每个村庄的路径，从一个村庄到达另一个，有一个且只有一条路径（地图绘制像一棵树）。

该公司计划在每个岛屿的其中一个村庄建立分工厂。所有在岛上不同地方挖来的5种稀有矿产将被发送到这个分工厂，后成为一个复合金属。因此途径的道路必须被重修过，才能通过数量庞大的车队。为了尽量减少对道路交通的建设成本，公司决定扩大原有的路径，而不是兴建新的道路。此外，该公司决定兴建村庄的矿田，以确保他们的工人可以进入采矿。 （如果子工厂坐落在一个村庄中，矿田也可以建在该村庄。）

由于这些岛屿的矿石储量非常高，每一种罕见的矿物只需要一矿田开采即可。鉴于目前所选择的岛屿的地图，你需要找到在每个岛上的建立子工厂的最优村庄，并从该村庄开始扩展道路，以采集所有5种稀有的矿产。

 

输入

第一行，包含一个整数T，表示岛屿的数目。（1<=Num<=50）

每组测试数据，第一行一个整数N（5<=N<=1000）。表示岛上村庄的个数。

接下来一行，N个数，m1, m2, ... mn(0<=mi<=5),表示第i个村庄所拥有的稀有矿藏的种类，0表示该村庄没有稀有矿藏（每一个村庄至多只有一种矿藏）。

接下来N-1行，描述了这个岛上连接N个村庄的地图。

每行三个数，x，y，d(1<=x, y<=n, 1<=d<=10000), 表示从x和y两个村庄之间有一条距离为d的路。

 

输出

每行输出一个整数。在每个岛上所需的最少扩建道路的长度。

 

样例输入

2

6

1 2 3 4 5 5

1 2 100

2 3 82

3 4 73

4 5 120

4 6 108

6

1 1 2 3 4 5

1 2 56

1 3 100

1 4 100

2 5 100

3 6 100

样例输出

363

456

Solution

首先这里有一棵树，可以考虑树形DP

又有五种矿藏，必须记录下来，想到了状压DP

F[i][j]表示以i为根结点的子树，收集到的矿藏为j时的最短路径

F[u][i|j]=min{f[u][i|j],f[u][i]+f[v][j]+dis[u][v]}

 

其中，u是v的父亲

要注意的是，u和v可以有重复的矿藏，这样方程写起来顺一些

#include<bits/stdc++.h> 
using namespace std; 
  
const int MAXN=1010,MAXE=2020; 
int head[MAXN],Next[MAXE],son[MAXE],dis[MAXE]; 
int f[MAXN][32]; 
bool flag[MAXN]; 
int tot,n; 
  
void add_e(int x,int y,int z) 
{ 
    tot++; 
    Next[tot]=head[x]; 
    head[x]=tot; 
    son[tot]=y; 
    dis[tot]=z;  
} 
  
void init() 
{ 
    int x,y,z; 
    scanf("%d",&n); 
    for (int i=1;i<=n;i++) 
    { 
        scanf("%d",&x); 
        f[i][1<<x-1]=0; 
    } 
    for (int i=1;i<n;i++) 
    { 
        scanf("%d%d%d",&x,&y,&z); 
        add_e(x,y,z); add_e(y,x,z); 
    } 
} 
  
void dfs(int u) 
{ 
    flag[u]=true; 
    for (int i=head[u];i;i=Next[i]) 
    { 
        int v=son[i]; 
        if (!flag[v]) 
        { 
            dfs(v); 
            for (int j=0;j<=31;j++) 
                for (int k=0;k<=31;k++) 
                    f[u][j|k]=min(f[u][j|k],f[u][j]+f[v][k]+dis[i]); 
            f[u][31]=min(f[u][31],f[v][31]);             
        } 
    } 
} 
  
int main() 
{ 
    int T; 
    scanf("%d",&T); 
    while (T--) 
    { 
        memset(f,0x3f,sizeof(f)); 
        memset(head,0,sizeof(head)); tot=0; 
        memset(flag,false,sizeof(flag)); 
        init(); 
        dfs(1);  
        printf("%d\n",f[1][31]); 
    } 
    return 0; 
}