cf1039D 分块

cf1039D

链接

cf

思路

一次k可以贪心O(n)算。
对于\(≤\sqrt{n}\)的k，暴力算。
对于\(＞\sqrt{n}\)的k，最多会有\(\sqrt{n}\)种答案，而且答案单调。
二分就行了。
复杂度\(O(nlogn+n\sqrt{n}logn)\)
递归会被卡，所以要记录dfs序然后循环

代码

#include 
#include 
#include 
using namespace std;
const int _=1e5+7;
inline int read() {
    int x=0,f=1;char s=getchar();
    for(;s>'9'||s<'0';s=getchar()) if(s=='-') f=-1;
    for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';
    return x*f;
}
int n,g[_],head[_],tot,stak[_],fa[_],top;
struct node {int v,nxt;}e[_<<1];
inline void add(int u,int v) {e[++tot].v=v,e[tot].nxt=head[u],head[u]=tot;}
void dfs(int u,int F) {
    fa[u]=F,stak[++top]=u;
    for(int i=head[u];i;i=e[i].nxt)
        if(e[i].v!=F) dfs(e[i].v,u);
}
int calc(int L) {
    int ans=0;
    for(int i=1;i<=n;++i) g[i]=1;
    for(int i=n;i>=2;--i) {
        int u=stak[i],v=fa[u];
        if (g[v]+g[u]>=L) ++ans,g[v]=0;
        else if(g[v]) g[v]=max(g[v],g[u]+1);
    }
    return ans;
}
int main() {
    n=read();
    for(int i=1,u,v;i>1;
            if(calc(mid)==ans) R=mid,l=mid+1;
            else r=mid-1;
        }
        for(;i<=R;++i) printf("%d\n",ans);
    }
    return 0;
}