CodeForces 95 E. Lucky Country（并查集+多重背包）

Description

定义全部由$4,7$组成的数组为幸运数字，给出一张$n$个点$m$条边的无向图，要求增加最少的边使得该图中有一个点数为幸运数字的连通块

Input

第一行两个整数$n,m$表示点数和边数，之后$m$行每行输入一条无向边$(1\le n,m\le 10^5)$

Output

输出最少需要加边数，如果无解则输出$-1$

Sample Input

4 3
1 2
2 3
1 3

Sample Output

1

Solution

并查集求出每个连通分支的点数后，问题转化为一个多重背包问题

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100005;
int n,m,fa[maxn],num[maxn],a[maxn],dp[maxn];
int find(int x)
{
    if(x==fa[x])return x;
    return fa[x]=find(fa[x]);
}
void unite(int x,int y)
{
    x=find(x),y=find(y);
    if(x==y)return ;
    fa[x]=y;
    num[y]+=num[x];
}
void ZeroOne_Pack(int cost,int weight) 
{
    for(int i=n;i>=cost;i--)
        dp[i]=min(dp[i],dp[i-cost]+weight);
}
void Multiple_Pack(int cost,int weight,int amount)
{
    int k=1;
    while(k<amount)
    {
        ZeroOne_Pack(k*cost,k*weight);
        amount-=k;
        k*=2;
    }
    ZeroOne_Pack(amount*cost,amount*weight);
}
bool check(int x)
{
    while(x)
    {
        if(x%10!=4&&x%10!=7)return 0;
        x/=10;
    }
    return 1;
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=1;i<=n;i++)fa[i]=i,num[i]=1;
        while(m--)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            unite(u,v);
        }
        for(int i=1;i<=n;i++)
            if(find(i)==i)a[num[i]]++;
        memset(dp,INF,sizeof(dp));
        dp[0]=0;
        for(int i=1;i<=n;i++)
            if(a[i])Multiple_Pack(i,1,a[i]);
        int ans=INF;
        for(int i=1;i<=n;i++)
            if(check(i))ans=min(ans,dp[i]);
        if(ans==INF)printf("-1\n");
        else printf("%d\n",ans-1);
    }
    return 0;
}