【NOIP模拟赛】林下风气

nzymeteor

于 2019-09-24 21:29:18 发布

阅读量224

点赞数 1

分类专栏：提高组文章标签： NOIP 模拟赛林下风气

本文链接：https://blog.csdn.net/nzymmwds/article/details/101314267

版权

提高组专栏收录该内容

54 篇文章 0 订阅

订阅专栏

在这里插入图片描述

题解：

dp维护乱搞，强行枚举最小值，然后算出联通块满足的值f[i][1]和不满足的值f[i][0];

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=3350,P=19260817;
int n,k,rt,num;
int a[N],last[N];
ll f[N][2],ans;
struct edge {
	int to,next;
} e[2*N];
void add(int x,int y) {
	e[++num].to=y;
	e[num].next=last[x];
	last[x]=num;
}
void dfs(int x,int fa) {
	if(a[x]==a[rt]+k) f[x][1]=1,f[x][0]=0;
	else f[x][1]=0,f[x][0]=1;
	for(int w=last[x]; w; w=e[w].next) {
		int y=e[w].to;
		if(y==fa||a[y]==a[rt]&&y>rt||a[y]<a[rt]||a[y]>a[rt]+k) continue;
		dfs(y,x);
		f[x][1]=(f[x][1]+f[x][1]*f[y][0]%P+f[x][0]*f[y][1]%P+f[x][1]*f[y][1]%P)%P;
		f[x][0]=(f[x][0]+f[x][0]*f[y][0]%P)%P;
	}
}
int read(){
	int num=0;
	char ch=getchar();
	while(ch>'9'||ch<'0'){
		ch=getchar();
	}
	while(ch>='0'&&ch<='9'){
		num=(num<<1)+(num<<3)+ch-'0';
		ch=getchar();
	}
	return num;
}
int main() {
	n=read(),k=read();
	for(int i=1;i<=n;i++) 
		a[i]=read();
	for(int i=1;i<n;i++){
		int x=read(),y=read();
		add(x,y),add(y,x);
	}
	for(int i=1;i<=n;i++) 
		rt=i,dfs(rt,0),ans=(ans+f[i][1])%P;
	printf("%lld",ans);
}