您的位置:首页 > 编程语言 > Java开发

UVALive 5717 Peach Blossom Spring(斯坦纳树 STNT)

2013-09-09 19:12 459 查看
题目大意:给你n个点,m条边,问你建立至少包含前k个点和后k个点的最小生成树,也可能是森林,每棵树的前面点的个数要和后面点的个数相等。

思路:STNT。如果大家对斯坦纳树不是很了解的,可以参看这里:http://endlesscount.blog.163.com/blog/static/821197872012525113427573/

这道题目还有点不一样,那就是它可能最后搞出来是森林,所以读入,先把斯坦纳树搞好,然后在做一遍DP,这遍DP比较简单,就是状态加加搞搞就好了,由于个数要一样,还要加一个check函数判断,即 d[ s ] = min(d[ s1 ] + d[ s - s ]),check(s) == 1 && check(s1) == 1。

第一次写STNT,那篇博客写的不错,有的地方想了挺长时间,但感觉还是没有很清楚的理解,再做几道吧~~

代码如下:

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

const int INF = 0x0fffffff;

const int MAX_N = 55;
const int MAX_M = 1111;

struct Edge
{
int t,next,w;
} edge[MAX_M<<1];

int tot,head[MAX_N];

void add_edge(int s,int t,int w)
{
edge[tot].t = t;
edge[tot].w = w;
edge[tot].next = head[s];
head[s] = tot++;
}

int d[1<<11][MAX_N];

int val[MAX_N];

void init(int n,int k)
{
int S = 1<<(k<<1);
for(int s = 0;s < S ;s ++)
for(int i = 1;i<=n;i++)
d[s][i] = INF;
memset(val,0,sizeof(val));
for(int i = 1;i<=k;i++)
{
val[i] = 1<<(i - 1);
val[n - i +1] = 1<<(2*k - i);
d[val[i]][i] = 0;
d[val[n - i + 1]][n - i + 1] = 0;
}
}

struct Node
{
int s,id;
Node(){}
Node(int a,int b) : s(a),id(b) {}
};

queue <Node> q;

int update(int s,int id,int w)
{
if(w < d[s][id])
{
d[s][id] = w;;
return 1;
}
else return 0;
}

bool inq[1<<11][MAX_N];

void spfa()
{
while(!q.empty())
{
int s = q.front().s;
int id = q.front().id;
q.pop();
inq[s][id] = 0;
for(int e = head[id];e != -1;e = edge[e].next)
{
int v = edge[e].t;
int w = edge[e].w;
if(update(s|val[v],v,d[s][id] + w) && s|val[v] ==s && !inq[s][v])//注意update,update要写在前面,第一个要更新
{
inq[s][v]= 1;
q.push(Node(s,v));
}
}
}
}

void stnt(int n,int k)
{
init(n,k);
memset(inq,0,sizeof(inq));
int S = 1<<(k<<1);
for(int s = 1;s<S;s++)
{
for(int i = 1;i<=n;i++)
{
for(int s1 = (s - 1)&s ;s1;s1 = (s1 - 1)&s)
{
int s2 = s - s1;
d[s][i] = min(d[s][i] , d[s1|val[i]][i] + d[s2|val[i]][i]);
}
d[s][i] = min(d[s][i],d[0][i] + d[s][i]);
if(d[s][i] < INF) q.push(Node(s,i)),inq[s][i] = 1;
}
spfa();
//for(int i = 1;i<=n;i++)
//printf("s = %d,i = %d,d = %d\n",s,i,d[s][i]);
}
}

int d2[1<<11];

int check(int s,int k)
{
int cnt1 = 0,cnt2 = 0;
for(int i = 0;i<k && s;i++)
{
if(s&1) cnt1 ++;
s >>= 1;
}
for(int i = 0;i<k && s;i++)
{
if(s&1) cnt2 ++;
s >>= 1;
}
return cnt1 == cnt2 ? 1 : 0 ;
}

int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n,m,k;
scanf("%d%d%d",&n,&m,&k);
tot=0;
memset(head,-1,sizeof(head));
for(int i = 1;i<=m;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
add_edge(a,b,c);
add_edge(b,a,c);
}

stnt(n,k);

int S = 1<<(k<<1);
for(int s = 1;s<S;s++)
{
d2[s] = INF;
for(int i = 1;i<=n;i++)
d2[s] = min(d2[s],d[s][i]);
}

for(int s = 1;s < S;s++)
{
if(check(s,k))
for(int s1 = (s - 1)&s;s1;s1 = (s1 - 1)&s)
{
if(check(s1,k))
{
int s2 = s - s1;
d2[s] = min(d2[s],d2[s1] + d2[s2]);
}
}
}
if(d2[S - 1] >= INF) puts("No solution");
else printf("%d\n",d2[S - 1]);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: