Description

Input

Output

Sample Input

【樣例1 輸入】

5 3 3
2 1 1 2 3
1 2
2 3
1 4
1 5

【樣例2 輸入】

10 2 3
2 1 1 3 1 2 3 3 1 3
1 2
2 3
2 4
2 5
2 6
5 7
1 8
8 9
1 10

Sample Output

【樣例1 輸出】

11

【樣例2 輸出】

435

Data Constraint

Hint

Solution

• 我們枚舉一個點為根，使它作為必選的第 K 大的點。

• 判斷以它為根為子樹中比它大的點的數(shù)量是否大于 k?1 ，滿足即可進(jìn)行樹形DP。

• 設(shè) f[i][j] 表示在以 i 為子樹中以 i 為第 j 大的方案數(shù)。

• 直接枚舉轉(zhuǎn)移即可，最后答案加上 d[i]?f[i][k] 。

• 暴力跑得真慢……

Code

#include<cstdio> #include<cstring> #include<cctype> using namespace std; const int N=1667,mo=64123; int tot,s,k,ans; int first[N],nex[N<<1],en[N<<1]; int d[N],f[N][N]; inline int read() {int X=0,w=0; char ch=0;while(!isdigit(ch)) w|=ch=='-',ch=getchar();while(isdigit(ch)) X=(X<<3)+(X<<1)+(ch^48),ch=getchar();return w?-X:X; } inline void insert(int x,int y) {nex[++tot]=first[x];first[x]=tot;en[tot]=y; } void dfs(int x,int y) {int pd=d[x]>d[s] || d[x]==d[s] && x>s;for(int i=1;i<=k;i++) f[x][i]=f[y][i-pd];for(int i=first[x];i;i=nex[i])if(en[i]^y) dfs(en[i],x);for(int i=1;i<=k;i++) f[y][i]=(f[y][i]+f[x][i])%mo; } int main() {int n=read();k=read();read();for(int i=1;i<=n;i++) d[i]=read();for(int i=1;i<n;i++){int x=read(),y=read();insert(x,y);insert(y,x);}for(int i=1;i<=n;i++){int tot=1;for(int j=1;j<i;j++) tot+=d[j]>d[i];for(int j=i+1;j<=n;j++) tot+=d[j]>=d[i];if(tot<k) continue;memset(f[i],0,sizeof(f[i]));f[s=i][1]=1;for(int j=first[i];j;j=nex[j]) dfs(en[j],i);ans=(ans+d[i]*f[i][k]%mo)%mo;}printf("%d",ans);return 0; }

總結(jié)

以上是生活随笔為你收集整理的JZOJ 5639. 【NOI2018模拟4.8】秘密袭击的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。