[CSP-S模拟测试]:Cover(单调栈+单调队列+DP)

题目传送门(内部题126)


输入格式

  第一行两个个整数$n,m$表示区间的长度与彩灯的数量。
  接下来$m$行,每行三个整数$l_i,r_i,a_i$表示一条彩灯能够覆盖的区间以及它的美观程度。


输出格式

  输出一行$m$个整数,第$i$个数表示$k=i$时的最大美观程度。


样例

样例输入:

25 6
1 2 10
2 3 10
1 3 21
3 4 10
4 5 10
3 5 19

样例输出:

41 80 80 80 80 80


数据范围与提示

  对于$25\%$的数据,$m\leqslant 20$
  对于$45\%$的数据,$n,m\leqslant 5,000$
  对于另外$25\%$的数据,所有$a_i$相同
  对于$100\%$的数据,$1\leqslant l_i\leqslant r_i\leqslant n,m\leqslant 300,000,a_i\leqslant 10^9$


题解

因为不能重叠,所以将所有的区间向其覆盖的区间连边,单调栈维护即可。

然后会得到一棵树,对于每一个节点维护一个单调队列更新其父节点答案即可,思想类似树上$DP$。

时间复杂度:$\Theta(n\log n)$。

期望得分:$100$分。

实际得分:$100$分。


代码时刻

#include<bits/stdc++.h>
using namespace std;
struct rec{int id,l,r,a;}s[400001];
struct node{int nxt,to;}e[700001];
int head[400001],cnt;
int n,m;
int now=2;
int size[300001];
long long ans;
priority_queue<long long> q[400001],v;
bool cmp(rec a,rec b){return a.l<b.l||(a.l==b.l&&a.r>b.r)||(a.l==b.l&&a.r==b.r&&a.id>b.id);}
void add(int x,int y)
{
	e[++cnt].nxt=head[x];
	e[cnt].to=y;
	head[x]=cnt;
}
void build(int x)
{
	while(s[x].l<=s[now].l&&s[now].r<=s[x].r)
	{
		add(x,now);
		now++;
		build(now-1);
	}
}
void dfs(int x)
{
	for(int i=head[x];i;i=e[i].nxt)
	{
		dfs(e[i].to);
		size[x]=max(size[x],size[e[i].to]);
	}
	size[x]++;
	for(int i=head[x];i;i=e[i].nxt)
		if(size[e[i].to]+1==size[x]){swap(q[x],q[e[i].to]);e[i].to=0;break;}
	for(int i=head[x];i;i=e[i].nxt)
	{
		if(e[i].to)
		{
			while(q[e[i].to].size())
			{
				v.push(q[x].top()+q[e[i].to].top());
				q[x].pop();q[e[i].to].pop();
			}
			swap(v,q[x]);
			while(v.size())
			{
				q[x].push(v.top());
				v.pop();
			}
		}
	}
	q[x].push(s[x].a);
}
int main()
{
	scanf("%d%d",&n,&m);n--;
	for(int i=1;i<=m;i++)
	{
		s[i].id=i;
		scanf("%d%d%d",&s[i].l,&s[i].r,&s[i].a);
		s[i].r--;
	}
	s[++m]=(rec){m,1,n,0};
	sort(s+1,s+m+1,cmp);
	build(1);dfs(1);
	for(int i=1;i<=m;i++)q[1].push(0);
	for(int i=1;i<m;i++)
	{
		ans+=q[1].top();
		q[1].pop();
		printf("%lld ",ans);
	}
	return 0;
}

rp++

posted @ 2019-11-07 17:48  HEOI-动动  阅读(189)  评论(0编辑  收藏  举报