【DP】 cf 486D
1216 단어 dp
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int mod = 1000000007;
const int maxn = 2005;
const int maxm = 4005;
struct Edge
{
int v;
Edge *next;
}E[maxm], *H[maxn], *edges;
LL f[maxn];
int a[maxn];
int n, m;
void addedges(int u, int v)
{
edges->v = v;
edges->next = H[u];
H[u] = edges++;
}
void init()
{
edges = E;
memset(H, 0, sizeof H);
}
void dfs(int u, int fa, int rt)
{
f[u] = 1;
for(Edge *e = H[u]; e; e = e->next) if(e->v != fa) {
int v = e->v;
if(a[v] < a[rt] || a[v] > a[rt] + m) continue;
if(a[v] == a[rt] && v < rt) continue;
dfs(v, u, rt);
f[u] = f[u] * (f[v] + 1) % mod;
}
}
void work()
{
scanf("%d%d", &m, &n);
for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
for(int i = 1; i < n; i++) {
int u, v;
scanf("%d%d", &u, &v);
addedges(u, v);
addedges(v, u);
}
LL ans = 0;
for(int i = 1; i <= n; i++) {
dfs(i, i, i);
ans = (ans + f[i]) % mod;
}
printf("%I64d
", ans);
}
int main()
{
//freopen("data", "r", stdin);
init();
work();
return 0;
}
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
【경쟁 프로 전형적인 90문】008의 해설(python)의 해설 기사입니다. 해설의 이미지를 봐도 모르는 (이해력이 부족한) 것이 많이 있었으므로, 나중에 다시 풀었을 때에 확인할 수 있도록 정리했습니다. ※순차적으로, 모든 문제의 해설 기사를 들어갈 예정입니다. 문자열...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.