HDU 6041 I Curse Myself

由于每条边最多只在一个环上，因此可以将环上的边权值，构成一个集合。问题就变成了，从若干个集合中，每个集合取一个数，计算K大的总和。

一次性考虑所有集合不太容易，可以将集合依次合并过去。那样复杂度就是O(KM)。

#include<bits/stdc++.h>
using namespace std;

const int MAXN=1010;
const int MAXM=4020;

struct Edge
{
	int x,y,v;
}e[MAXM];

struct opt
{
	int w,x,y;
	bool operator <(const opt& R)const
	{
		return w<R.w;
	}
};

vector<int> vec[MAXN],w,arr;
vector<int>::iterator it;
int K;
int vis[MAXN],pre[MAXN],vise[MAXM];

void merge(vector<int> &v,vector<int> b)
{
	priority_queue <opt> q;
	for(int i=0;i<b.size();i++)
		q.push((opt{b[i]+v[0],i,0}));
	w.resize(0);
	while(w.size()<=K&&!q.empty())
	{
		opt tmp=q.top();
		q.pop();
		w.push_back(tmp.w);
		if(tmp.y+1<v.size())
		{
			tmp.y++;
			q.push((opt){b[tmp.x]+v[tmp.y],tmp.x,tmp.y});
		} 
	}
	v=w;
}

void dfs(int x)
{
	int i,son,tmp,nd;
	vis[x]++;
	for(i=0;i<vec[x].size();i++)
	{
		if(vise[vec[x][i]])
			continue;
		vise[vec[x][i]]++;
		son=e[vec[x][i]].x^e[vec[x][i]].y^x;
		if(vis[son])
		{
			vis[x]=2;
			tmp=pre[x];
			nd=x;
			vector<int> arrtmp;
			arrtmp.push_back(e[vec[x][i]].v);
			while(1)
			{
				arrtmp.push_back(e[tmp].v);
				nd=e[tmp].x^e[tmp].y^nd;
				if(nd==son)
					break; 
				tmp=pre[nd];
			}
			merge(arr,arrtmp);
		}
		else
		{
			pre[son]=vec[x][i];
			dfs(son);
		}
	}
}

int main()
{
	arr.reserve(100100);
	w.reserve(100100);
	int as,sum;
	int n,m,i,j,cas=1;
	while(~scanf("%d%d",&n,&m))
	{
		for(i=0;i<=n;i++)
			vec[i].resize(0);
		arr.resize(0);
		arr.push_back(0);
		sum=0;
		for(i=0;i<m;i++)
		{
			scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].v);
			vec[e[i].x].push_back(i);
			vec[e[i].y].push_back(i);
			sum+=e[i].v;
		}
		scanf("%d",&K);
		memset(vis,0,sizeof(vis));
		memset(vise,0,sizeof(vise));
		memset(pre,-1,sizeof(pre));
		dfs(1);
		as=0;
		sort(arr.begin(),arr.end());
		for(i=0;i<arr.size()&&i<K;i++)
			as+=(sum-arr[arr.size()-1-i])*(i+1);
		printf("Case #%d: %u\n",cas++,as);
	}
}