ABC 309 E Family and Insurance

Linear_L發表於2024-06-03

題意
一個家庭用一顆樹來表示。其中有m個人買了保險,x[i]買的保險可以繼承y[i]代,請問有多少人至少有一份保險

思路
感覺是比較水的E題了,我們採取bfs遍歷,然後類似於最短路的想法來更新每個點可以繼承的最大保險代數。最後掃一遍所有人,看他們的dis有多少大於等於0,即為答案(dis最初所有人賦值為-1),具體的看程式碼吧沒啥好說的。

程式碼

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=3e5+10;
vector<int>ds[maxn];
int dis[maxn];
bool vis[maxn];

struct node
{
	int id;
	int dis;
};

bool operator <(node a,node b)
{
	return a.dis<b.dis;
}

signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);
	
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++) dis[i]=-1;
	for(int i=2;i<=n;i++) 
	{
		int fa;
		cin>>fa;
		ds[fa].push_back(i);
	}
	priority_queue<node>q;
	for(int i=1;i<=m;i++)
	{
		int x,y;
		cin>>x>>y; 
		q.push((node){x,y});
		dis[x]=max(dis[x],y);
	}
	while(!q.empty())
	{
		node term=q.top();
		q.pop();
		int u=term.id;
		if(vis[u]) continue;
		vis[u]=1;
		for(int i=0;i<ds[u].size();i++)
		{
			int y=ds[u][i];
			if(dis[y]<dis[u]-1) 
			{
				dis[y]=dis[u]-1;
				if(dis[y]&&!vis[y]) q.push((node){y,dis[y]});
			}
		}
	}
	int much=0;
	for(int i=1;i<=n;i++) if(dis[i]>=0) much++;
	cout<<much<<endl;
	return 0;
}