洛谷4365 [九省联考2018]秘密袭击coat

标签：DP，搜索

题目

题目传送门

题目背景

We could have had it all… …

我们本该，拥有一切

Counting on a tree… …

何至于此，数数树上

Counting on a Tree（CoaT）即是本题的英文名称。

题目描述

Access Globe 最近正在玩一款战略游戏。在游戏中，他操控的角色是一名C 国士
兵。他的任务就是服从指挥官的指令参加战斗，并在战斗中取胜。

C 国即将向D 国发动一场秘密袭击。作战计划是这样的：选择D 国的s 个城市，
派出C 国战绩最高的s 个士兵分别秘密潜入这些城市。每个城市都有一个危险程度$d_i$，

C 国指挥官会派遣战绩最高的士兵潜入所选择的城市中危险程度最高的城市，派遣战绩第二高的士兵潜入所选择的城市中危险程度次高的城市，以此类推（即派遣战绩第i高的士兵潜入所选择城市中危险程度第i 高的城市）。D 国有n 个城市，n - 1 条双向道路连接着这些城市，使得这些城市两两之间都可以互相到达。为了任务执行顺利，C 国选出的s 个城市中，任意两个所选的城市，都可以不经过未被选择的城市互相到达。

Access Globe 操控的士兵的战绩是第k 高，他希望能估计出最终自己潜入的城市的
危险程度。Access Globe 假设C 国是以等概率选出任意满足条件的城市集合S ，他希望你帮他求出所有可能的城市集合中，Access Globe 操控的士兵潜入城市的危险程度之和。如果选择的城市不足k 个，那么Access Globe 不会被派出，这种情况下危险程度为0。

当然，你并不想帮他解决这个问题，你也不打算告诉他这个值除以998 244 353 的
余数，你只打算告诉他这个值除以64,123 的余数。

输入输出格式

输入格式

从文件coat.in 中读入数据。

第1 行包含3 个整数n、k、W，表示D 国城市的个数、Access Globe 所操控士兵
潜入的城市战绩排名以及D 国的所有城市中最大的危险程度；

第2 行包含n 个1 到W 之间的整数$d_1$; $d_2$; … $d_n$，表示每个城市的危险程度；

第3 行到第n + 1 行，每行两个整数$x_i$; $y_i$，表示D 国存在一条连接城市$x_i$ 和城市$y_i$ 的双向道路。

输出格式

输出到文件coat.out 中。
输出一个整数，表示所有可行的城市集合中，Access Globe 操控的士兵潜入城市的危险程度之和除以64,123 的余数。

输入输出样例

输入样例#1

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

输出样例#1

11

输入样例#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

输出样例#2

435

说明

D 国地图如下，其中危险程度为d 的城市的形状是(d + 3) 边形。

以下是所有符合条件且选择的城市不少于3 个的方案：

• 选择城市1、2、3，Access Globe 的士兵潜入的城市危险程度为1；

• 选择城市1、2、3、4，Access Globe 的士兵潜入的城市危险程度为1；

• 选择城市1、2、3、5，Access Globe 的士兵潜入的城市危险程度为1；

• 选择城市1、2、3、4、5，Access Globe 的士兵潜入的城市危险程度为2；

• 选择城市1、2、4，Access Globe 的士兵潜入的城市危险程度为1；

• 选择城市1、2、5，Access Globe 的士兵潜入的城市危险程度为1；

• 选择城市1、2、4、5，Access Globe 的士兵潜入的城市危险程度为2；

• 选择城市1、4、5，Access Globe 的士兵潜入的城市危险程度为2；而在选择的
城市少于3 时，Access Globe 的士兵潜入的城市危险程度均为0；

所以你应该输出(1 + 1 + 1 + 2 + 1 + 1 + 2 + 2) mod 64 123 = 11。

以下是所有符合条件且选择的城市不少于3 个的方案：

• 选择城市1、2、3，Access Globe 的士兵潜入的城市危险程度为1；

• 选择城市1、2、3、4，Access Globe 的士兵潜入的城市危险程度为1；

• 选择城市1、2、3、5，Access Globe 的士兵潜入的城市危险程度为1；

• 选择城市1、2、3、4、5，Access Globe 的士兵潜入的城市危险程度为2；

• 选择城市1、2、4，Access Globe 的士兵潜入的城市危险程度为1；

• 选择城市1、2、5，Access Globe 的士兵潜入的城市危险程度为1；

• 选择城市1、2、4、5，Access Globe 的士兵潜入的城市危险程度为2；

• 选择城市1、4、5，Access Globe 的士兵潜入的城市危险程度为2；而在选择的 城市少于3 时，Access Globe 的士兵潜入的城市危险程度均为0；

所以你应该输出(1 + 1 + 1 + 2 + 1 + 1 + 2 + 2) mod 64 123 = 11。

分析

枚举每个可能的点，算出以它为第k名的联通块个数

将它作为根，按照dfs序DP

f[x][k]表示以x节点为第k名的联通块个数

注意每次要清空f[x]

code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#define mem(x,num) memset(x,num,sizeof x)
#define reg(x) for(int i=last[x];i;i=e[i].next)
using namespace std;
inline ll read(){
    ll f=1,x=0;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
//**********head by yjjr**********
const int maxn=5e3+6,mod=64123;
int n,k,W,cnt,S,ans,last[maxn],f[maxn][maxn],d[maxn];
struct edge{int to,next;}e[maxn<<1];
void insert(int u,int v){
    e[++cnt]=(edge){v,last[u]};last[u]=cnt;
    e[++cnt]=(edge){u,last[v]};last[v]=cnt;
}
void dfs(int x,int fa,int g[]){
    if(d[x]>d[S]||(d[x]==d[S]&&x>S))rep(i,1,k)f[x][i]=g[i-1];
    else rep(i,1,k)f[x][i]=g[i];
    reg(x)if(e[i].to!=fa)dfs(e[i].to,x,f[x]);
    rep(i,1,k)g[i]=(g[i]+f[x][i])%mod;
}
void calc(int x){
    int tot=1;
    rep(i,1,n)if(d[i]>d[x]||(d[i]==d[x]&&i>x))tot++;//规定相等的点只能从小走到大的，防止重复 
    if(tot<k)return;
    S=x;mem(f[x],0);f[x][1]=1;
    reg(x)dfs(e[i].to,x,f[x]);
    ans=(ans+1ll*d[x]*f[x][k])%mod;
}
int main()
{
    n=read(),k=read(),W=read();
    rep(i,1,n)d[i]=read();
    rep(i,1,n-1){
        int u=read(),v=read();
        insert(u,v);
    }
    rep(i,1,n)calc(i);//枚举每个点作为根 
    cout<<ans<<endl;
    return 0;
}