Codeforces Round 888 (Div. 3)E. Nastya and Potions(搜索）

#include<bits/stdc++.h>
using namespace std;
int dx[4]={0,0,-1,1};
int dy[4]={-1,1,0,0};
#define int long long
typedef pair<int,int> pii;
const int mod=1e9+7;

const int N=200010;

int a[N];
vector<int>g[N];
int st[N];

int dfs(int x)
{
	if(st[x])return a[x];
	st[x]=1;
	if(g[x].size()==0)return a[x];
	int sum=0;
	for(int i=0;i<g[x].size();i++)
	{
		int t=g[x][i];
		dfs(t);//应该先搜索，更新了孩子的信息后再进行加和 
		sum+=a[t];
	}
	a[x]=min(a[x],sum);
	return a[x];
}
void solve()
{
	int n,k;
	cin>>n>>k;
	memset(st,0,sizeof st);
	for(int i=1;i<=n;i++)cin>>a[i];
	for(int i=1;i<=k;i++)
	{
		int x;cin>>x;
		a[x]=0;
	} 
	for(int i=1;i<=n;i++)
	{
		g[i].clear();
		int k1;
		cin>>k1;
		while(k1--)
		{
			int x;
			cin>>x;
			g[i].push_back(x);
		}
	} 
	for(int i=1;i<=n;i++)
	{
		dfs(i);
	}
	for(int i=1;i<=n;i++)cout<<a[i]<<" ";
	cout<<endl; 
} 
signed main()
{ 
    int good_luck_to_you;
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    good_luck_to_you=1;
    cin>>good_luck_to_you;
    while(good_luck_to_you--)
    {
       solve();
    }
    system("pause");   
}
         

 思路就是dfs,记录这道题的原因是因为我把dfs的顺序给放错了，导致更新的信息很乱，这道题相当是要先更新孩子的信息，然后再更新父节点的信息，所以在更新信息时应该放在dfs的后面。

牛客周赛 Round 50 小红的树上移动

#include<bits/stdc++.h>
using namespace std;
int dx[4]={0,0,-1,1};
int dy[4]={-1,1,0,0};
#define int long long
const int mod=998244353;
typedef pair<int,int> pii;

const int N=1000010;

vector<int>g[N];
int n;
int d[N],a[N],b[N];
int p[N];
int maxndepth;

int qmi(int a,int b,int mod)
{
    int res=1;
    while(b)
    {
        if(b&1)res=(res*a)%mod;
        b>>=1;
        a=(a*a)%mod;
    }
    return res;
}
void dfs(int x,int fa)
{
    d[x]=d[fa]+1;
    maxndepth=max(maxndepth,d[x]);
    a[d[x]]++;
    if(g[x].size()==1&&x!=1)
    {
        b[d[x]]++;
    }
    for(int i=0;i<g[x].size();i++)
    {
        int t=g[x][i];
        if(t==fa)continue;
        dfs(t,x);
    }
}
void solve()
{
	cin>>n;
	int m=n-1;
	while(m--)
	{
		int a,b;
		cin>>a>>b;
		g[a].push_back(b);
        g[b].push_back(a);
	}
    d[0]=0;
    dfs(1,0);
    int sum=0;
    p[1]=1;
    for(int i=2;i<=maxndepth;i++)
    {
        p[i]=p[i-1]*(a[i]-b[i])%mod*qmi(a[i],mod-2,mod)%mod;
    }
    for(int i=1;i<=maxndepth;i++)
    {
        sum+=p[i];
        sum%=mod;
    }
	cout<<sum<<endl;
} 
signed main()
{ 
    int good_luck_to_you;
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    good_luck_to_you=1;
  // cin>>good_luck_to_you;
    while(good_luck_to_you--)
    {
       solve();
    }
    system("pause");   
}
         

本题的思路还是很清晰的，就是移动到同一层，如果是叶子节点就不再移动，那么我们要记录每层有几个结点和叶子节点，每层的期望就是非叶子节点的期望，注意除法取模要用逆元来求。