一道非常好的最小生成树题目 。

看似非常复杂,事实上细致分析一下算法的复杂度就会发现。假设增加了lrj说的优化。事实上复杂度不高 。

就像紫书中说的, 除去购买套餐中的点,剩下的最小边仍然在原始的最小生成树中 。  所以我们用二进制枚举子集的方法枚举全部购买套餐的组合。然后将套餐中的点增加并查集中,再用原始最小生成树中的边补全当前生成树 。

二进制枚举子集的复杂度是2^8 。

补全生成树的复杂度是O(n) 。 所以最后复杂度为O(n*(2^8)) 。约等于10^6 。能够接受。

有一个地方我不知道是不是坑啊。题目中明明说两点间的欧几里得距离的平方是费用。可是这不是一个浮点数吗?为什么用int接收这个值反而是对的呢?希望知道答案的朋友能指教 。

细节參见代码:

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1000 + 5;
const int maxm = maxn*(maxn);
int T,n,m,rec,q,cnt,par[maxn], x[maxn],y[maxn];
struct node {
    int n,c,a[maxn];
}p[9];
struct edge{
    int a,b;
    int v;
}ed[maxm],e[maxn];
bool cmp(edge a,edge b) {
    return a.v < b.v;
}
int find(int x) { return par[x] == x ? x : par[x] = find(par[x]); }
int solve() {
    for(int i=1;i<=n;i++) par[i] = i;
    sort(ed,ed+cnt,cmp);
    int res = 1 ,ans = 0;
    rec = 0;
    for(int i=0;i<cnt;i++) {  //拿出原始最小生成树的边集
        int x = find(ed[i].a) , y = find(ed[i].b);
        if(x != y) {
            ans += ed[i].v;
            e[rec].a = ed[i].a;
            e[rec].b = ed[i].b;
            e[rec++].v = ed[i].v;
            par[x] = y; res++;
        }
        if(res == n) break;
    }
    for(int s=0;s<(1<<q);s++) {    //二进制枚举全部套餐的可能情况
        for(int j=1;j<=n;j++) par[j] = j;  //初始化并查集
        int cur = 0;
        for(int j=0;j<q;j++) {
            if(s & (1<<j)) {
                cur += p[j].c;
                for(int i=1;i<=p[j].n;i++) {
                    int x = find(p[j].a[i]) , y = find(p[j].a[1]);
                    if(x != y)
                        par[x] = y;
                }
            }
        }
        for(int i=0;i<rec;i++) { //补边
            int x = find(e[i].a) , y = find(e[i].b);
            if(x != y) {
                cur += e[i].v;
                par[x] = y;
            }
        }
        ans = min(ans,cur); //更新
    }
    return ans;

}
int main() {
    scanf("%d",&T);
    while(T--) {
        scanf("%d%d",&n,&q);
        for(int i=0;i<q;i++) {
            scanf("%d%d",&p[i].n,&p[i].c);
            for(int j=1;j<=p[i].n;j++) scanf("%d",&p[i].a[j]);
        }
        for(int i=1;i<=n;i++) scanf("%d%d",&x[i],&y[i]);
        cnt = 0;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++) {
                int v = (x[i]-x[j])*(x[i]-x[j]) + (y[i]-y[j])*(y[i]-y[j]);
                ed[cnt].a = i; ed[cnt].b = j; ed[cnt++].v = v;
            }
        printf("%d\n",solve());
        if(T) printf("\n");
    }
    return 0;
}


版权声明:本文为博主原创文章,未经博主同意不得转载。 举报
  • 本文已收录于下面专栏:

相关文章推荐

UVA 1151 Buy or Build(最小生成树+枚举子集)

题目地址:https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem...

Should we buy it now or wait for better features and lower price?

&lt;!-- [if gte mso 9]&gt;&lt;xml&gt; &lt;w:WordDocument&gt; &lt;w:View&gt;Normal&lt;/w:View&gt; &lt;w:Zoom&gt;0&lt;/w:Zoom&gt; &lt;w:PunctuationKerning/&gt; &lt;w:DrawingGridVerticalSpacing&gt;7.8 磅&lt;/w:DrawingGridVerticalSpacing&gt; &lt;w:DisplayHorizontalDrawingGridEvery&gt;
  • wsql
  • wsql
  • 2010-12-08 11:29
  • 306

UVA 1151 Buy or Build 最小生成树+二进制选取子集

题目链接:点击打开链接 题意:给你n个点,你的任务是让这n个点连通。

为此,你有两种方法,1、在某两点之间建边,费用为两点之间欧几里得距离的平方。2、购买一些套餐。当买了某个套餐后,套餐中的这些点将变...

uva 1151 Buy or Build (最小生成树)

uva 1151 Buy or Build题目大意:给出n个点以及m种套餐,每种套餐的数据包括改套餐所包括点的数量,该套餐的费用。以及改套餐所包括的详细的点的编号。点与点之间的费用是他们的欧几里得距离...

Buy or Build UVA - 1151 MST最小生成树

紫书P358 收获: 1:用vector建立须要的边,和原本就有的边 2:这里的subn用vector事实上是非常正常的,可是我可能原本想不出要用这个 3:掩码mask在这里的表示要更加熟练 ...

Should we buy it now or wait for better features and lower price?

&lt;!-- [if gte mso 9]&gt;&lt;xml&gt; &lt;w:WordDocument&gt; &lt;w:View&gt;Normal&lt;/w:View&gt; &lt;w:Zoom&gt;0&lt;/w:Zoom&gt; &lt;w:PunctuationKerning/&gt; &lt;w:DrawingGridVerticalSpacing&gt;7.8 磅&lt;/w:DrawingGridVerticalSpacing&gt; &lt;w:DisplayHorizontalDrawingGridEvery&gt;

UVA 1151 Buy or Build (最小生成树)

题意: 告诉你n 个点的坐标。你要在两个点之间连线,使得点所有同样。连边的费用为这两个点的欧几里得距离。你的目的是使这个费用最低,而且你有q(q 思路: 所有的点相通,非常明显是最小生成树。 ...

POJ1151+线段树+扫描线

/* 线段树+扫描线+离散化 求多个矩形的面积 */ #include&lt;stdio.h&gt; #include&lt;string.h&gt; #include&lt;stdlib.h&gt; #include&lt;algorithm&gt; #include&lt;iostream&gt; #include&lt;queue&gt; #include&lt;stack&gt; #include&lt;math.h&gt; #include&lt;map&gt; using namespace std; const int m
  • 微博
    微信
    QQ
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多仅仅同意输入30个字)