Poj(2784),二進制枚舉+最小生成樹kruskal Buy or Build

題目鏈接:https://vjudge.net/contest/179332#problem/I

心得:&在C語言中的運算,如果運算對象有兩個,那麼&表示位與運算。結果中的每一個二進制位等於兩個運算數的對應位置的二進制位安位與。每一個位的位於運算法則是,當且僅當運算數都爲1時結果爲1,即:1&1==1,1&0==0&1==0。例如:

代碼:

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long LL;
struct node{
	LL x;
	LL y;
	LL w;
}edge[1000010];
LL T,n,q;
LL buy[1005];
LL ans=0;
LL set[1005];
vector<LL>pile[10];//相當於pile[i][10]
LL ant;
LL cmp(node a,node b)
{
	return a.w<b.w;
}
void retur()
{
	for(LL i=0;i<=n;i++)
	{
		set[i]=i;
	}
}
LL find(LL a)
{//此處用遞歸實現就AC 用while實現就Time limit exceeded
 //不知道爲什麼 
	if(a!=set[a])
        set[a] = find(set[a]);
    return set[a];
/*    while(a!=set[a])
    {
    	a=set[a];
	}
	return a;*/
}

LL kruskal()
{
	LL i;	
	ans=0;
	LL o=0;
	for(i=0;i<ant;i++)
	{
		LL a=edge[i].x;
		LL b=edge[i].y;
		if(find(a)!=find(b))
		{
			set[find(a)]=find(b);
			ans=ans+edge[i].w;
			o++;
		}
		if(o==n-1)
		{
			break;
		}
	}
}
LL min(LL a,LL b)
{
	if(a>b)
	{
		a=b;
	}
	return a;
}
int main()
{
	scanf("%lld",&T);
	while(T--)
	{
		LL i,j;
		scanf("%lld%lld",&n,&q);
		for(i=0;i<q;i++)
		{
			pile[i].clear();
			LL k;
			scanf("%lld%lld",&k,&buy[i]);
			for(j=0;j<k;j++)
			{
				LL pot;
				scanf("%lld",&pot);
				pile[i].push_back(pot);
			}
		}
		LL x1[1005],y1[1005];	
		for(i=1;i<=n;i++)
		{
			scanf("%lld%lld",&x1[i],&y1[i]);
		}
		ant=0;
		for(i=1;i<=n;i++)
		{
			for(j=i+1;j<=n;j++)
			{
				edge[ant].x=i;
				edge[ant].y=j;
				edge[ant].w=(x1[i]-x1[j])*(x1[i]-x1[j])+(y1[i]-y1[j])*(y1[i]-y1[j]);
				ant++;
			}
		}
		sort(edge,edge+ant,cmp);
		retur();
		kruskal();
		LL result=ans;
		
		LL mark;
		LL sum;
		for(mark=1;mark<(1<<q);mark++)//二進制枚舉 
		{
			retur();
			sum=0;
			for(j=0;j<q;j++)
			{
				if(mark&(1<<j))
				{
					sum=sum+buy[j];
					for(i=1;i<pile[j].size();i++)
					{
						LL a=pile[j][i];
						LL b=pile[j][0];
						if(find(a)!=find(b))
						{
							set[find(a)]=find(b);
						}
					} 
				}
			}
			kruskal();
			result=min(result,sum+ans);
		}
		printf("%lld\n",result);
		memset(x1,0,sizeof(x1));
		memset(y1,0,sizeof(y1));
		if(T)
		{
			printf("\n");
		}
	}	
	return 0;	
}


發佈了63 篇原創文章 · 獲贊 16 · 訪問量 1萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章