트리 dp
5219 단어 dpcodeforces트리 DP
제목 링크:
codeforces 161D
제목 대의:
나무 한 그루를 주십시오. 각 변의 변권은 1입니다. 두 점 사이의 경로 길이가 k인 점 쌍은 몇 개입니까?
제목 분석:
dp[u][j]=∑v∈Vdp[v][j−1]
dp[u][j]+=dp[p][j−1]−dp[u][j−2]
AC 코드:
#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <algorithm>
#define MAX 50007
using namespace std;
typedef long long LL;
int n,k,a,b;
LL dp[MAX][507],ans;
vector<int> e[MAX];
void add ( int u , int v )
{
e[u].push_back ( v );
e[v].push_back ( u );
}
void Clear ( )
{
for ( int i = 0 ; i < MAX ; i++ )
e[i].clear();
}
void dfs ( int u , int p )
{
dp[u][0] = 1;
for ( int i = 1 ; i <= k ; i++ )
dp[u][i] = 0;
for ( int i = 0 ; i < e[u].size() ; i++ )
{
int v = e[u][i];
if ( v == p ) continue;
dfs ( v , u );
for ( int j = 1 ; j <= k ; j++ )
dp[u][j] += dp[v][j-1];
}
}
void solve ( int u , int p )
{
for ( int i = 0 ; i < e[u].size() ; i++ )
{
int v = e[u][i];
if ( v == p ) continue;
for ( int j = k; j >= 1 ; j-- )
{
dp[v][j] += dp[u][j-1];
if ( j > 1 ) dp[v][j] -= dp[v][j-2];
}
solve ( v , u );
}
}
int main ( )
{
while ( ~scanf ( "%d%d" , &n , &k ) )
{
Clear();
for ( int i = 1 ; i < n ; i++ )
{
scanf ( "%d%d" , &a , &b );
add ( a , b );
}
ans = 0;
dfs ( 1 , -1 );
solve ( 1 , -1 );
/*for ( int i = 1; i <= n ; i++ ) for ( int j = 0 ; j <= k ; j++ ) cout << i << " " << j << " " << dp[i][j] << endl;*/
for ( int i = 1 ; i <= n ; i++ )
ans += dp[i][k];
printf ( "%I64d
" , ans/2LL );
}
}
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
【경쟁 프로 전형적인 90문】008의 해설(python)의 해설 기사입니다. 해설의 이미지를 봐도 모르는 (이해력이 부족한) 것이 많이 있었으므로, 나중에 다시 풀었을 때에 확인할 수 있도록 정리했습니다. ※순차적으로, 모든 문제의 해설 기사를 들어갈 예정입니다. 문자열...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.