Xenon's Attack on the Gangs Codeforces Round #614 (Div. 2)

Xenon’s Attack on the Gangs
题意: 给你一棵树,将0~n-2一一赋值给n-1条边,则S最大可能取值
S = ∑ 1 ≤ u < v ≤ n m e x ( u , v ) S=\sum_{1\le u< v\le n}mex(u,v) S=1u<vnmex(u,v)
mex(u,v)表示u到v的路径中未出现的最小非负数
思路: 这里推荐一个视频题解,个人觉得讲得非常好 qscqesze
S = ∑ 1 ≤ u < v ≤ n m e x ( u , v ) = ∑ 1 ≤ x ≤ n ( ∑ m e x ( u , v ) = x x ) = ∑ 1 ≤ x ≤ n ( ∑ m e x ( u , v ) ≥ x 1 ) S=\sum_{1\le u< v\le n}mex(u,v)=\sum_{1\le x\le n}(\sum_{mex(u,v)=x}x)=\sum_{1\le x \le n}(\sum_{mex(u,v)\ge x}1) S=1u<vnmex(u,v)=1xn(mex(u,v)=xx)=1xn(mex(u,v)x1)
所以我们只需要考虑每条边的贡献即可,每条边要有贡献。
首先我们可以看0所在的边的两个端点u,v,则0的贡献为size[v][u]*size[u][v],接着看1,我们发现只有将0,1连接在一起贡献才会尽可能的大,所以我们要使得[0,m]分布在一条链上。所以我们可以对这棵树进行dp,dp[u][v]表示以u,v为端点的链

#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<stack>
#include<queue>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int N=3e3+5;

vector<int>G[N];
int fa[N][N],sz[N][N];
ll dp[N][N],ans;

void dfs(int u,int f,int root){
    sz[root][u]=1;
    fa[root][u]=f;
    for(auto v: G[u]){
        if(v==f)
            continue;
        dfs(v,u,root);
        sz[root][u]+=sz[root][v];
    }
}

ll sl(int x,int y){
    if(x==y)
        return 0;
    if(dp[x][y]!=-1)
        return dp[x][y];
    dp[x][y]=1ll*sz[y][x]*sz[x][y]+max(sl(fa[y][x],y),sl(x,fa[x][y]));
    return dp[x][y];
}

int main(){
#ifdef Mizp
    freopen("in.txt","r",stdin);
#endif
    std::ios::sync_with_stdio(0);
    int n;
    cin>>n;
    for(int i=1;i<n;i++){
        int u,v;
        cin>>u>>v;
        G[u].push_back(v);
        G[v].push_back(u);
    }

    for(int i=1;i<=n;i++){
        dfs(i,0,i);
    }
    memset(dp,-1,sizeof dp);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            ans=max(ans,sl(i,j));
        }
    }
    cout<<ans<<endl;
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值