CF1119F Niyaz and Small Degrees【treedp+堆】

如果枚举d来dp,那么就是设f[u][0/1]为u点不断/断掉和父亲的边,然后优先选取f[v][1]+w(u,v)<=f[v][0]的,如果断掉这些度数还是多就用一个堆维护剩下的按f[v][1]+w(u,v)-f[v][0]从大到小取(负的)
然后注意到设当前要求度数是d,那么度数<=d的点全都不用dp,所以可以直接把贡献累加到和他相邻的点上然后断掉边,这样一共算Σ度数ci,就是O(nlogn)的了

#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>
#include<algorithm>
using namespace std;
const int N=300005;
int n,du,v[N],d[N],s[N],ne[N];
long long ans,sm[N],f[N][2];
vector<pair<int,int> >e[N];
vector<int>a[N];
int read()
{
	int r=0,f=1;
	char p=getchar();
	while(p>'9'||p<'0')
	{
		if(p=='-')
			f=-1;
		p=getchar();
	}
	while(p>='0'&&p<='9')
	{
		r=r*10+p-48;
		p=getchar();
	}
	return r*f;
}
struct dui
{
	priority_queue<long long>p,q;
	void push(long long x)
	{
		q.push(x);
	}
	void erase(long long x)
	{
		p.push(x);
	}
	int top()
	{
		while(!q.empty()&&!p.empty()&&q.top()==p.top())
			q.pop(),p.pop();
		return q.top();
	}
	void pop()
	{
		top();
		q.pop();
	}
	int size()
	{
		return q.size()-p.size();
	}
	bool empty()
	{
		return q.size()==p.size();
	}
}q[N];
bool cmp(const pair<int,int>&a,const pair<int,int>&b)
{
	return d[a.first]<d[b.first];
}
void add(int u,int v,int w)
{
	e[u].push_back(make_pair(v,w));
	e[v].push_back(make_pair(u,w));
}
void dfs(int u,int fa)
{
	v[u]=1;
	int r=d[u]-du;
	long long sum=0;
	vector<long long>ad,de;
	while(q[u].size()>r)
		sm[u]-=q[u].top(),q[u].pop();
	while(s[u]<e[u].size()&&d[e[u][s[u]].first]<=du)
		s[u]++;
	for(int i=s[u],len=e[u].size();i<len;i++)
		if(!v[e[u][i].first]&&e[u][i].first!=fa)
		{
			dfs(e[u][i].first,u);
			if(f[e[u][i].first][1]+e[u][i].second<=f[e[u][i].first][0])
				r--,sum+=f[e[u][i].first][1]+e[u][i].second;
			else
			{
				sum+=f[e[u][i].first][0];
				long long nw=f[e[u][i].first][1]+e[u][i].second-f[e[u][i].first][0];
				q[u].push(nw);
				sm[u]+=nw;
				de.push_back(nw);
			}
		}
		while(q[u].size()>max(0,r))
			sm[u]-=q[u].top(),ad.push_back(q[u].top()),q[u].pop();
		f[u][0]=sum+sm[u];
		r--;
		while(q[u].size()>max(0,r))
			sm[u]-=q[u].top(),ad.push_back(q[u].top()),q[u].pop();
		f[u][1]=sum+sm[u];
		for(int i=0,len=ad.size();i<len;i++)
			q[u].push(ad[i]),sm[u]+=ad[i];
		for(int i=0,len=de.size();i<len;i++)
			q[u].erase(de[i]),sm[u]-=de[i];
}
int main()
{
	n=read();
	for(int i=1;i<n;i++)
	{
		int x=read(),y=read(),z=read();
		add(x,y,z);
		d[x]++,d[y]++;
		ans+=z;
	}
	for(int i=1;i<=n;i++)
	{
		a[d[i]].push_back(i);
		sort(e[i].begin(),e[i].end(),cmp);
	}
	ne[n]=n+1;
	for(int i=n-1;i>=1;i--)
	{
		if(a[i+1].size())
			ne[i]=i+1;
		else
			ne[i]=ne[i+1];
	}
	printf("%lld ",ans);
	for(int i=1;i<n;i++)
	{
		for(int j=0,len=a[i].size(),u;j<len;j++)
		{
			u=a[i][j];
			v[u]=1;
			for(int k=0,le=e[u].size();k<le;k++)
				if(!v[e[u][k].first])
					q[e[u][k].first].push(e[u][k].second),sm[e[u][k].first]+=e[u][k].second;
		}
		ans=0,du=i;
		for(int j=i+1;j<=n;j=ne[j])
			for(int k=0,len=a[j].size();k<len;k++)
				if(!v[a[j][k]])
				{
					dfs(a[j][k],0);
					ans+=f[a[j][k]][0];
				}
		for(int j=i+1;j<=n;j=ne[j])
			for(int k=0,len=a[j].size();k<len;k++)
				v[a[j][k]]=0;
		printf("%lld ",ans);
	}
	return 0;
}
posted @ 2019-05-11 10:26  lokiii  阅读(274)  评论(0编辑  收藏  举报