Codeforces 1249F最大重量サブセット（貪欲）

問題の意味

ツリーの少し右側では、その結果、これらの点は二十から二距離kより大きく、最大点と右、複数の点から選択されます。

考え

貪欲に比較的大きな値をとり
、現在のビット右場合、降順に深度ソート内の全ての点\（[I] \） K内のすべての点について減少距離次いで、0より大きい\（A [I ] \）
代表的な答え貢献として、この点を取った[I] \）（\
ゼロよりも右も大きい以下と点までスイープ場合ポイントがこれより大きいことを示し、次に取る
の複雑さを\（O（N ^ 2 ）\）

コード

int n,k;
int a[maxn],b[maxn];
vector<int>v[maxn];
int ans;
int dep[maxn];
void dfs(int x, int fa, int dp){
    dep[x]=dp;
    for(int i = 0; i < (int)v[x].size(); i++){
        int y = v[x][i];
        if(y==fa)continue;
        dfs(y,x,dp+1);
    }
}
int gao(int st){
    queue<PI>q;
    q.push(make_pair(st,0));
    int C = a[st];
    while(!q.empty()){
        int x = q.front().fst;
        a[x]-=C;
        int stp = q.front().sc;
        q.pop();
        if(stp!=k){
            for(int i = 0; i < (int)v[x].size(); i++){
                int y = v[x][i];
                q.push(make_pair(y,stp+1));
            }
        }
    }
    return C;
}
bool cmp(int a, int b){return dep[a]>dep[b];}
int main(){
    scanf("%d %d", &n, &k);
    for(int i = 1; i <= n; i++){
        scanf("%d", &a[i]);b[i]=i;
    }
    for(int i = 1; i < n; i++){
        int x,y;
        scanf("%d %d", &x, &y);
        v[x].pb(y);v[y].pb(x);
    }dfs(1,-1,0);
    sort(b+1,b+1+n,cmp);
    for(int i = 1; i <= n; i++){
        if(a[b[i]]>0)ans+=gao(b[i]);
    }
    printf("%d\n",ans);
    return 0;
}