码迷,mamicode.com
首页 > 其他好文 > 详细

分珠(dfs+并查集)

时间:2016-07-20 22:57:35      阅读:162      评论:0      收藏:0      [点我收藏+]

标签:

1140 分珠

时间限制:500MS  内存限制:65536K
提交次数:24 通过次数:18

题型: 编程题   语言: G++;GCC

 

Description

如下图所示,有若干珠子,每颗珠子重量不同,珠子之间有一些细线将它们连在一起。现要求切断一些细线,将它们分成两部分,
分割后,单独每一部分的珠子仍保持相连,且要求尽量做到两部分总重相等或相差最少。
请编一程序,给定珠子个数、每颗珠子的重量以及珠子之间的连接情况,输出按上述要求分割后两部分总重的差值的绝对值。




输入格式

第一行有两个数N与M(1<=N,M<=10),N为珠子个数(珠子编号依次为1,2,3,...,N),M为连接珠子的细线数目。第二行为N个正整数,分别为N个珠子的重量。此后M行,每行两个数X与Y,表示珠子X与珠子Y由细线相连。



输出格式

按要求分割后两部分总重的差值的绝对值。



 

输入样例

5 5
1 2 3 4 1
1 2
1 3
2 3
3 4
4 5



 

输出样例

1



 

题解

  dfs枚举出所有去掉某些边的情况。去掉边后,用并查集求出此时有多少个联通块。如果正好有两个联通块,则更新最小值

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
typedef pair <int,int> pii;
const int maxn=20;
pii P[maxn];
int n,m,cur,ans,val[maxn],flag[maxn],sum[maxn],M[maxn][maxn],f[maxn];
int find(int x)
{
    int r=x,i=x,t;
    while (r!=f[r]) r=f[r];
    while (i!=r)//路径压缩
    {
        t=f[i];
        f[i]=r;
        i=t;
    }
    return r;
}
void mix(int x,int y)
{
    int fx=find(x),fy=find(y);
    if (fx!=fy)
    {
        f[fx]=fy;
        sum[fy]+=sum[fx];//将儿子的值加给祖先
        cur--;//有两个块合并,联通块数量减一
    }
}
void init()
{
    for (int i=1;i<=n;i++)
    sum[i]=val[i],f[i]=i;
    cur=n;//联通块数量,初始值为n
}
int getnum()
{
    init();
    for (int i=0;i<m;++i)
    {
        if (flag[i])//该边没被取消
        mix(P[i].first,P[i].second);
    }
    if (cur==2)
    {
        for (int i=2;i<=n;i++)
        if (find(i)!=find(1))//此时只有两个联通块,可以直接找出与1不是同一个祖先的
        return abs(sum[find(1)]-sum[find(i)]);
    }
    return -1;//联通块数量不为2
}
void dfs(int pos)
{
    if (pos==n+1)
    return ;
    flag[pos]=0;//取消该边
    int temp=getnum();
    if (temp==-1)//得不到结果
        dfs(pos+1);
    else ans=min(ans,temp);//得到了结果,就没必要再取消这条边的情况下继续递归下去了,直接更新
    flag[pos]=1;
    dfs(pos+1);//在不取消该边的情况下递归
}
int main()
{
    scanf("%d%d",&n,&m);
    ans=0;
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&val[i]);
        ans+=val[i];
    }
    int a,b;
    for (int i=0;i<m;i++)
    {
        scanf("%d%d",&a,&b);
        P[i]=make_pair(a,b);
        flag[i]=1;
    }
    dfs(1);
    printf("%d\n",ans);
    return 0;
}

 

分珠(dfs+并查集)

标签:

原文地址:http://www.cnblogs.com/scaugsh/p/5689814.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!