4006: [JLOI2015]管道连接

题目链接

题目大意：有一些关键点，每个点有一个颜色，使任意同色的关键点两两联通，求最小代价

题解：首先斯坦纳树一下，然后再对不同颜色状压一次

我的收获：2333

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;

#define INF 0x3f3f3f3f

const int N=1077;
const int M=10005;

int n,m,K;
int p[N],c[N],d[N],has[N];
int t,head[N];

struct edge{int to,val,nex;}e[M];
void add(int u,int v,int w){e[t].to=v,e[t].val=w,e[t].nex=head[u],head[u]=t++;}

queue<int> q;

struct SteinerTree{

int cnt,all;
int f[N][N],G[N];
bool vis[N];

void input()
{
    t=0;memset(head,-1,sizeof(head));
    scanf("%d%d%d",&n,&m,&K);all=1<<K;
    for(int i=1;i<=K;i++) p[i]=1<<i-1;
    while(m--){  
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z),add(y,x,z);
    }
    for(int i=1;i<=K;i++) scanf("%d%d",&c[i],&d[i]);
    for(int i=1;i<=K;i++) has[c[i]]|=p[i];//按输入顺序对d[i]编号1~10 
    memset(f,0x3f,sizeof(f));
    for(int i=1;i<=K;i++) f[d[i]][p[i]]=0;
}

void spfa(int state)
{
    while(!q.empty())
    {
        int u=q.front();q.pop();vis[u]=0;
        for(int i=head[u];i!=-1;i=e[i].nex){
            int v=e[i].to;
            if(f[v][state]>f[u][state]+e[i].val)
            {
                f[v][state]=f[u][state]+e[i].val;
                if(!vis[v]) q.push(v),vis[v]=1;
            }
        }
    }
}

void dp()
{
    for(int i=0;i<all;i++){
        for(int u=1;u<=n;u++)
        {
            for(int s=i;s;s=(s-1)&i)
                 f[u][i]=min(f[u][i],f[u][s]+f[u][i^s]);
            if(f[u][i]<INF) q.push(u),vis[u]=1;
        }
        spfa(i); 
    }
}

void calc()
{
    memset(G,0x3f,sizeof(G));
    for(int i=0;i<all;i++){
        int now=0;
        for(int j=1;j<=10;j++) if(i&p[j]) now|=has[j];
        for(int j=1;j<=n;j++)  
            G[now]=min(G[now],f[j][now]);
    }
    for(int i=0;i<all;i++){
        for(int s=i;s;s=(s-1)&i)
            G[i]=min(G[i],G[s]+G[i^s]);  
    }
    cout<<G[all-1]<<endl;
}

void work()
{
    dp();
    calc();
}

}T;

int main()
{
    T.input();
    T.work();
    return 0;
}