[4교 합동훈련] 나무 자르기 게임. - 나무 DP. - 조합 수학.
5137 단어 동적 기획수론......수론적 기초......트리 DP
사실 접두사 적과 접두사 적을 적으면 돼요. 역원을 요구할 필요 없어요.
#include
#include
#include
#define R register
#define ll long long
#define max_n 100010
#define mod 1000000007
using namespace std;
struct ED{int to,nex;}edge[200100];
int n,et;
int col[max_n],st[max_n];
ll f[max_n][2];
int read()
{
R int xx;R char ch;
while(ch=getchar(),ch<'0'||ch>'9');xx=ch-'0';
while(ch=getchar(),ch>='0'&&ch<='9')xx=xx*10+ch-'0';
return xx;
}
void exgcd(ll &x,ll &y,ll a,ll b)
{
if(b==0)x=1,y=0;
else
{
exgcd(x,y,b,a%b);
ll tmp=x;
x=(y+mod)%mod;
y=(tmp-(a/b)*y%mod+mod)%mod;
}
}
void dfs(int u,int pre)
{
R int e,v;
R ll x,y;
R ll sum1=0,sum2=1;
for(e=st[u];e!=-1;e=edge[e].nex)
if((v=edge[e].to)!=pre)
{
dfs(v,u);
sum2=sum2*(f[v][1]+f[v][0])%mod;
}
if(col[u]==1){f[u][1]=sum2,f[u][0]=0;return;}
for(e=st[u];e!=-1;e=edge[e].nex)
if((v=edge[e].to)!=pre)
{
exgcd(x,y,(f[v][0]+f[v][1])%mod,mod);
sum1=(sum1+f[v][1]*(sum2*x%mod)%mod)%mod;
}
f[u][1]=sum1%mod,f[u][0]=sum2%mod;
}
int main()
{
R int i,j;
R int u,v;
n=read();
memset(st,-1,sizeof(st));
et=0;
for(i=1;i<=n;++i)col[i]=read();
for(i=1;iread(),v=read();
edge[et]=(ED){v,st[u]},st[u]=et++;
edge[et]=(ED){u,st[v]},st[v]=et++;
}
dfs(1,1);
printf("%lld",f[1][1]);
return 0;
}
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
01 가방, 완전 가방, 다중 가방 dp(동적 기획 입문 dp)01 가방은 2진법으로 직접 표시할 수 있지만 데이터 양이 너무 많으면 시간을 초과하는 것이 폭력이다.01 가방의 사상은 바로 이 물품에 대해 내가 넣은 가치가 큰지 안 넣은 가치가 큰지 비교하여 방정식 f[i][v...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.