SDOI2013 R1 Day1


2018.3.22 Test

时间:7:30~11:40
得分:80+65+20=165

总结

时间分配。。
2个月前写的都忘的差不多了。。
注意:T2合并一棵子树,这棵子树某些点的dep[x]可能会变小,于是fa[x][..]应变为0,但如果在循环里写 &&(1<<i)<=dep[x] 这个dep[x]是更新后的,fa[x][..]不会被更新成0,还是以前的fa[x][..],于是就挂了。

T1 BZOJ.3122.[SDOI2013]随机数生成器(BSGS 等比数列)

题目链接

想象假装自己学过的必修五数列。。
\(X_i = A*X_{i-1}+B (mod\ P)\),可以化成等差数列(构造一个项使得右边的B能进到括号里,即要减去一个B,且系数要*A,那么左右两边同时 \(+\frac{B}{A-1}\))
由等差数列第n项公式可得 \(X_n + \frac{B}{A-1} = A^{n-1}*(X_0 + \frac{B}{A-1}) (mod\ P)\)
只有A的次数是不知道的,把右边乘(除)过来就是BSGS了
要注意这个式子在 A=1 或 A=0 时是无意义的,要特判。

算出答案后先取模再+1(随便模什么。。)
我tm可不是sb -> 特判输出2也要模P。。666

//2788kb    336ms
#include <cmath>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define gc() getchar()
#define mod (100005)
typedef long long LL;
const int N=1e5+7;

LL P,A,B,X,pur;

struct Hash
{
    int H[N],Enum,nxt[N],to[N],res[N],val[N];//数组适当小一些 在清空的情况下更好 
    void Init(){
        Enum=0, memset(H,0,sizeof H);
    }
    void AddEdge(int id,int v)
    {
        int x=v%mod;
        for(int i=H[x]; i; i=nxt[i])
            if(val[i]==v) {to[i]=id; return;}
        to[++Enum]=id, nxt[Enum]=H[x], H[x]=Enum, val[Enum]=v;
    }
    int Query(int v)
    {
        for(int i=H[v%mod]; i; i=nxt[i])
            if(val[i]==v) return to[i];
        return -1;
    }
}hs;
inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now;
}
int gcd(int x,int y){
    return y?gcd(y,x%y):x;
}
void Ex_gcd(int a,int b,LL &x,LL &y){
    if(!b) x=1ll,y=0ll;
    else Ex_gcd(b,a%b,y,x),y-=a/b*x;
}
LL inv(int v,int p){
    LL x,y; Ex_gcd(v,p,x,y);
    return (x%p+p)%p;
}
LL FP(LL x,int k)
{
    LL t=1;
    for(; k; k>>=1,x=x*x%P)
        if(k&1) t=x*t%P;
    return t;
}
int BSGS()//A^x = pur (mod P)
{
    pur=(pur+B*inv(A-1,P)%P)*inv(X+B*inv(A-1,P)%P,P)%P;
    if(gcd(A,P)!=1) return -1;
    hs.Init();
    int m=ceil(sqrt(P-1));
    LL t=pur,AM;
    for(int i=0; i<=m; ++i,t=t*A%P) hs.AddEdge(i,t);
    t=AM=FP(A,m);
    for(int res,i=1; i<=m; ++i,t=t*AM%P)
        if((res=hs.Query(t))!=-1) return (1ll*i*m-res)%P+1ll;
    return -1;
}

int main()
{
    int T=read();
    while(T--)
    {
        P=read(),A=read(),B=read(),X=read(),pur=read();
        if(pur==X) printf("1\n");
        else if(!A){//这个只是判B!x_i = 0*x_{i-1}+B = B 
            printf("%d\n",B==pur?2:-1);
        }
        else if(A==1){//X_1+x*B = pur(mod p) -> x = (pur-X_1)*inv(B) (mod p)
            if((pur=(pur-X+P)%P)%gcd(B,P)) printf("-1\n");
            else printf("%d\n",(int)(pur*inv(B,P)%P+1ll));
        }
        else printf("%d\n",BSGS());
    }
    return 0;
}

T2 BZOJ.3123.[SDOI2013]森林(主席树 启发式合并)

题目链接

查树上第k小可以用主席树,但是合并。。(LCT再维护一个值的Splay行不行。。更改fa,son就Delete,Insert...<-作死的节奏)
合并两棵树我们可以用启发式合并,logn是可以的。
合并时还需要动态维护并上去的子树的LCA,所以用ST表求LCA,每次DFS一个点更新ST表即可。
合并时建主席树和之前DFS建树时是一样的,直接由父节点的root再加上A[x]。但这样每次合并一个点需要新开一个节点。。

//481804kb  8920ms
#include <cstdio>
#include <cctype>
#include <algorithm>
#define gc() getchar()
const int N=8e4+5,B=16;

int n,m,Fa[N],root[N],cnt,ref[N],A[N],dep[N],fa[N][17],sz[N],Enum,H[N],to[N<<1],nxt[N<<1];
struct Seg_Tree
{
    #define lson son[x][0]
    #define rson son[x][1]

    int tot,sum[N*500],son[N*500][2];//sum[N*18],son[N*18][2];
    void Insert(int x,int &y,int l,int r,int p)
    {
        sum[y=++tot]=sum[x]+1;
        if(l<r){
            int m=l+r>>1;
            if(p<=m) son[y][1]=rson,Insert(lson,son[y][0],l,m,p);
            else son[y][0]=lson,Insert(rson,son[y][1],m+1,r,p);
        }
    }
    int Query(int x,int y,int lca,int falca,int l,int r,int k)
    {
        if(l==r) return l;
        int delta=sum[lson]+sum[son[y][0]]-sum[son[lca][0]]-sum[son[falca][0]];
        if(delta>=k) return Query(lson,son[y][0],son[lca][0],son[falca][0],l,l+r>>1,k);
        return Query(rson,son[y][1],son[lca][1],son[falca][1],(l+r>>1)+1,r,k-delta);
    }
}t;

inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now;
}
inline void AddEdge(int u,int v)
{
    to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
    to[++Enum]=u, nxt[Enum]=H[v], H[v]=Enum;
}
int Find(int x)
{
    int l=1,r=cnt,mid;
    while(l<r)
        if(ref[mid=l+r>>1]<x) l=mid+1;
        else r=mid;
    return l;
}
void Discrete()
{
    std::sort(ref+1,ref+1+n);
    cnt=1;
    for(int i=2; i<=n; ++i)
        if(ref[i]!=ref[i-1]) ref[++cnt]=ref[i];
    for(int i=1; i<=n; ++i) A[i]=Find(A[i]);
}
void pre_DFS(int x,int anc)
{
    t.Insert(root[fa[x][0]],root[x],1,cnt,A[x]), Fa[x]=anc, ++sz[anc];
    for(int i=1; i<=B&&(1<<i)<=dep[x]; ++i)
        fa[x][i]=fa[fa[x][i-1]][i-1];
    for(int v,i=H[x]; i; i=nxt[i])
        if((v=to[i])!=fa[x][0])
            dep[v]=dep[x]+1, fa[v][0]=x, pre_DFS(v,anc);//, sz[x]+=sz[v];
}
int LCA(int x,int y)
{
    if(dep[x]<dep[y]) std::swap(x,y);
    int t=dep[x]-dep[y];
    for(int i=B; i>=0; --i)
//      if(dep[fa[x][i]]>=dep[x]) x=fa[x][i];
        if(t&(1<<i)) x=fa[x][i];
    if(x==y) return x;
    for(int i=B; i>=0; --i)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
void DFS(int x)
{
    t.Insert(root[fa[x][0]],root[x],1,cnt,A[x]);
    for(int i=1; i<=B/*&&(1<<i)<=dep[x]*/; ++i)//这个不能写!
        fa[x][i]=fa[fa[x][i-1]][i-1];
    for(int v,i=H[x]; i; i=nxt[i])
        if((v=to[i])!=fa[x][0])
            dep[v]=dep[x]+1, fa[v][0]=x, DFS(v);
}
int Get_fa(int x){
    return x==Fa[x]?x:Fa[x]=Get_fa(Fa[x]);
}
void Union(int x,int y)
{
    int r1=Get_fa(x),r2=Get_fa(y);
    if(sz[r1]>sz[r2]) std::swap(x,y),std::swap(r1,r2);
    Fa[r1]=r2, sz[r2]+=sz[r1];
    fa[x][0]=y, dep[x]=dep[y]+1, DFS(x);
}

int main()
{
//  freopen("forest.in","r",stdin);
//  freopen("forest.out","w",stdout);

//  int size = 256 << 20; // 256MB  
//  char *p = (char*)malloc(size) + size;  
//  __asm__("movl %0, %%esp\n" :: "r"(p));

    int test=read(),T;
    n=read(),m=read(),T=read();
    for(int i=1; i<=n; ++i) ref[i]=A[i]=read();
    Discrete();
    for(int u,v,i=1; i<=m; ++i) u=read(),v=read(),AddEdge(u,v);
    for(int i=1; i<=n; ++i) if(!dep[i]) pre_DFS(i,i);
    int ans=0,x,y,w,k; char opt[3];
    while(T--)
    {
        scanf("%s",opt), x=read()^ans, y=read()^ans;
        if(opt[0]=='Q') k=read()^ans, w=LCA(x,y),printf("%d\n",ans=ref[t.Query(root[x],root[y],root[w],root[fa[w][0]],1,cnt,k)]);
        else AddEdge(x,y), Union(x,y);
    }
    return 0;
}

T3 BZOJ.3124.[SDOI2013]直径(树 直径)

题目链接

需要判断一条边是否是所有直径都经过的边
可以证明(想象),树上的多条直径一定是从某个(些)点出发,中间经过同一条链(为方便把链上的边叫做关键边好了),再于某个(些)点结束。
只需要判断当前求出直径上的所有边是否为关键边。如果一条边(u->v)不是关键边,那么说明它(直径上)的前驱u,除了v之外还有u->v'可以满足最长(得到直径),于是对每个点记录子节点的最远(直径)、次远距离(非严格,在另一棵子树上)就好了。
如果是从R1出发,这样我们可以判掉R2那边的非关键边,从R2倒着再一遍即可判掉R1,得到中间那条链。

//17660kb   1932ms
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define gc() getchar()
typedef long long LL;
const int N=2e5+5;
const LL INF=1e10;

int n,Ans,root,Enum,H[N],to[N<<1],nxt[N<<1],fa[N];
LL Max,dis[N],mx[N],s_mx[N],val[N<<1];
bool is_d[N],Mark[N];

inline int read()
{
    int now=0,f=1;register char c=gc();
    for(;!isdigit(c);c=gc()) if(c=='-') f=-1;
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now*f;
}
inline void AddEdge(int u,int v,int w)
{
    to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum, val[Enum]=w;
    to[++Enum]=u, nxt[Enum]=H[v], H[v]=Enum, val[Enum]=w;
}
void pre_DFS(int x,LL d)
{
    if(d>Max) Max=d, root=x;
    for(int i=H[x]; i; i=nxt[i])
        if(to[i]!=fa[x]) fa[to[i]]=x,pre_DFS(to[i],d+val[i]);
}
void DFS(int x,int f)
{
    for(int v,i=H[x]; i; i=nxt[i])
        if((v=to[i])!=f)
        {
            DFS(v,x);
            if(mx[x]<val[i]+mx[v]) mx[x]=val[i]+mx[v];
            else if(s_mx[x]<val[i]+mx[v]) s_mx[x]=val[i]+mx[v];
        }
}
void DFS_for_Ans(int x,int f,LL d)
{
    for(int i=H[x]; i; i=nxt[i])
            if(to[i]!=f&&is_d[to[i]])
                if(s_mx[x]+d>=Max) {Mark[x]=1/*在此断开*/; return;}
    for(int i=H[x]; i; i=nxt[i])
        if(to[i]!=f&&is_d[to[i]])
            DFS_for_Ans(to[i],x,d+val[i]);
}

int main()
{
//  freopen("diameter.in","r",stdin);
//  freopen("diameter.out","w",stdout);

    n=read();
    for(int u,v,w,i=1; i<n; ++i) u=read(),v=read(),w=read(),AddEdge(u,v,w);
    Max=-INF, pre_DFS(1,0);
    int u=root,v;
    Max=-INF, fa[u]=0, pre_DFS(u,0), v=root;
    while(root!=u) is_d[root]=1,root=fa[root];
    is_d[root]=1;

    DFS(u,-1), DFS_for_Ans(u,-1,0);
    memset(mx,0,sizeof mx), memset(s_mx,0,sizeof s_mx);
    DFS(v,-1), DFS_for_Ans(v,-1,0);
    for(bool f=0; v!=u; v=fa[v],Ans+=f)
        if(Mark[v]) f^=1;
    printf("%I64d\n%d",Max,Ans);

    return 0;
}

考试代码

T1

#include <cmath>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define gc() getchar()
#define mod (100005)
typedef long long LL;
const int N=1e6+7;

LL P,A,B,X,pur;
bool vis[500];

struct Hash
{
    int H[N],Enum,nxt[N],to[N],res[N],val[N];
    void Init(){
        Enum=0, memset(H,0,sizeof H);
    }
    void AddEdge(int id,int v)
    {
        int x=v%mod;
        for(int i=H[x]; i; i=nxt[i])
            if(val[i]==v) {to[i]=id; return;}
        to[++Enum]=id, nxt[Enum]=H[x], H[x]=Enum, val[Enum]=v;
    }
    int Query(int v)
    {
        for(int i=H[v%mod]; i; i=nxt[i])
            if(val[i]==v) return to[i];
        return -1;
    }
}hs;
inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now;
}
int Spec1()
{
    memset(vis,0,sizeof vis);
    for(int tm=1; !vis[X]; X=(A*X+B)%P,++tm)
        if(X==pur) return tm;
        else vis[X]=1;
    return -1;
}
int gcd(int x,int y){
    return y?gcd(y,x%y):x;
}
void Ex_gcd(int a,int b,int &g,LL &x,LL &y){
    if(!b) x=1ll,y=0ll,g=a;
    else Ex_gcd(b,a%b,g,y,x),y-=a/b*x;
}
int Spec2()
{
    if(X==pur) return 1;
    int R=(pur-X%P+P)%P,gcd; LL x,y;
    Ex_gcd(B,P,gcd,x,y);
    if(R%gcd) return -1;
    x=1ll*x*(R/gcd)%P;
    P/=gcd;
    return (x%P+P)%P+1;
}
LL inv(int v,int p){
    LL x,y;int g; Ex_gcd(v,p,g,x,y);
    return (x%p+p)%p;
}
LL FP(LL x,int k)
{
    LL t=1;
    for(; k; k>>=1,x=x*x%P)
        if(k&1) t=x*t%P;
    return t;
}
int BSGS()
{
    if(gcd(A,P)!=1) return -2;
    hs.Init();
    int m=ceil(sqrt(P-1));
    LL t=pur%P,AM;
    for(int i=0; i<=m; ++i,t=t*A%P) hs.AddEdge(i,t);
    t=AM=FP(A,m);
    for(int res,i=1; i<=m; ++i,t=t*AM%P)
        if((res=hs.Query(t))!=-1) return (1ll*i*m-res)%P;
    return -2;
}
int Spec3()
{
    if(X==pur) return 1;
    pur=pur*inv(X,P)%P;
    return BSGS()+1;
}

int main()
{
    freopen("random.in","r",stdin);
    freopen("random.out","w",stdout);

    int T=read();
    while(T--)
    {
        P=read(),A=read(),B=read(),X=read()%P,pur=read();
        if(P<=100) printf("%d\n",Spec1());
        else if(A==1) printf("%d\n",Spec2());
        else if(!B) printf("%d\n",Spec3());
    }
    return 0;
}/*
1 9 1 6 1 4
*/

T2

样例LCT哪RE不知道。。没改
过220行了!写过最长的代码(虽然LCT部分没什么用)

#include <cstdio>
#include <cctype>
#include <algorithm>
#define gc() getchar()
const int N=8e4+5;

int n,m;
inline int read()
{
    int now=0,f=1;register char c=gc();
    for(;!isdigit(c);c=gc()) if(c=='-') f=-1;
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now*f;
}
#define lson son[x][0]
#define rson son[x][1]
namespace LCT
{
    int fa[N],son[N][2],val[N],sz[N],sk[N];
    bool tag[N];
    inline void Update(int x){
        sz[x]=sz[lson]+sz[rson]+1;
    }
    inline bool n_root(int x){
        return son[fa[x]][0]==x||son[fa[x]][1]==x;
    }
    inline void Rev(int x){
        std::swap(lson,rson), tag[x]^=1;
    }
    inline void PushDown(int x){
        if(tag[x]) Rev(lson),Rev(rson),tag[x]=0;
    }
    void Rotate(int x)
    {
        int a=fa[x],b=fa[a],l=son[a][1]==x,r=l^1;
        if(n_root(a)) son[b][son[b][1]==a]=x;
        if(son[x][r]) fa[son[x][r]]=a;
        fa[a]=x, fa[x]=b, son[a][l]=son[x][r], son[x][r]=a;
        Update(a);
    }
    void Splay(int x)
    {
        int t=1,a=x,b; sk[1]=x;
        while(n_root(a)) sk[++t]=a=fa[a];
        while(t) PushDown(sk[t--]);
        while(n_root(x))
        {
            a=fa[x], b=fa[a];
            if(n_root(a)) Rotate(son[a][1]==x^son[b][1]==a?x:a);
            Rotate(x);
        }
        Update(x);
    }
    void Access(int x){
        for(int pre=0; x; x=fa[pre=x])
            Splay(x), rson=pre, Update(x);
    }
    void Make_root(int x){
        Access(x), Splay(x), Rev(x);
    }
    void Split(int x,int y){
        Make_root(x), Access(y), Splay(y);
    }
    void Link(int x,int y){
        Make_root(x), fa[x]=y;
    }
}
namespace Spec
{
    int root[N],cnt,A[N],ref[N],Enum,H[N],to[N<<1],nxt[N<<1],top[N],dep[N],sz[N],fa[N],sn[N];
    bool vis[N];
    inline void AddEdge(int u,int v)
    {
        to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
        to[++Enum]=u, nxt[Enum]=H[v], H[v]=Enum;
    }
    struct Seg_tree
    {
        int tot,sum[N*18],son[N*18][2];
        void Insert(int x,int &y,int l,int r,int p)
        {
            sum[y=++tot]=sum[x]+1;
            if(l<r)
            {
                int m=l+r>>1;
                if(p<=m) son[y][1]=rson, Insert(lson,son[y][0],l,m,p);
                else son[y][0]=lson, Insert(rson,son[y][1],m+1,r,p);
            }
        }
        int Query(int x,int y,int z,int w,int l,int r,int k)
        {
            if(l==r) return l;
            int delta=sum[lson]+sum[son[y][0]]-sum[son[z][0]]-sum[son[w][0]];
            if(delta>=k) return Query(lson,son[y][0],son[z][0],son[w][0],l,l+r>>1,k);
            return Query(rson,son[y][1],son[z][1],son[w][1],(l+r>>1)+1,r,k-delta);
        }
    }t;
    int Find(int x,int r)
    {
        int l=1,mid;
        while(l<r)
            if(ref[mid=l+r>>1]>=x) r=mid;
            else l=mid+1;
        return l;
    }
    void Discrete()
    {
        cnt=1;
        for(int i=2; i<=n; ++i)
            if(ref[i]!=ref[i-1]) ref[++cnt]=ref[i];
        for(int i=1; i<=n; ++i) A[i]=Find(A[i],cnt);
    }
    void DFS1(int x)
    {
        int mx=0; sz[x]=1;
        for(int v,i=H[x]; i; i=nxt[i])
            if((v=to[i])!=fa[x])
            {
                dep[v]=dep[x]+1, DFS1(v), sz[x]+=sz[v];
                if(sz[v]>mx) mx=sz[v],sn[x]=v;
            }
    }
    void DFS2(int x,int tp)
    {
        top[x]=tp;
        if(sn[x])
        {
            DFS2(sn[x],tp);
            for(int i=H[x]; i; i=nxt[i])
                if(to[i]!=fa[x]&&to[i]!=sn[x]) DFS2(to[i],to[i]);
        }
    }
    void DFS(int x)
    {
        t.Insert(root[fa[x]],root[x],1,cnt,A[x]), vis[x]=1;
        for(int i=H[x]; i; i=nxt[i])
            if(to[i]!=fa[x]) fa[to[i]]=x, DFS(to[i]);
    }
    int LCA(int u,int v)
    {
        while(top[u]!=top[v])
        {
            if(dep[top[u]]<dep[top[v]]) std::swap(u,v);
            u=fa[top[u]];
        }
        return dep[u]<dep[v]?u:v;
    }
    void Solve(int T)
    {
        for(int i=1; i<=n; ++i) ref[i]=A[i]=read();
        std::sort(ref+1,ref+1+n), Discrete();
        for(int u,v,i=1; i<=m; ++i) u=read(),v=read(),AddEdge(u,v);
        for(int i=1; i<=n; ++i) if(!vis[i]) DFS(i);
        for(int i=1; i<=n; ++i) if(!sz[i]) DFS1(i),DFS2(i,i);
        int ans=0,x,y,k,w; char opt[3];
        while(T--)
        {
            scanf("%s",opt), x=read()^ans, y=read()^ans,k=read()^ans;
            w=LCA(x,y), printf("%d\n",ans=ref[t.Query(root[x],root[y],root[w],root[fa[w]],1,cnt,k)]);
        }
    }
}
namespace Violence
{
    int A[500],Enum,H[500],to[1000],nxt[1000],tmp[1000],pre[1000];
    bool vis[500];
    inline void AddEdge(int u,int v)
    {
        to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
        to[++Enum]=u, nxt[Enum]=H[v], H[v]=Enum;
    }
    void DFS(int x,int y,int f)
    {
        if(x!=y)
            for(int i=H[x]; i; i=nxt[i])
                if(to[i]!=f) pre[to[i]]=x, DFS(to[i],y,x);
    }
    int Query(int x,int y,int k)
    {
        DFS(x,y,-1);
        tmp[0]=0;
        while(y!=x) tmp[++tmp[0]]=A[y],y=pre[y];
        tmp[++tmp[0]]=A[x];
        std::sort(tmp+1,tmp+1+tmp[0]);
        return tmp[k];
    }
    void Solve(int T)
    {
        for(int i=1; i<=n; ++i) A[i]=read();
        for(int u,v,i=1; i<=m; ++i) u=read(),v=read(),AddEdge(u,v);
        int ans=0,x,y,k; char opt[3];
        while(T--)
        {
            scanf("%s",opt), x=read()^ans, y=read()^ans;
            if(opt[0]=='Q') k=read()^ans, printf("%d\n",ans=Query(x,y,k));
            else AddEdge(x,y);
        }
    }
}

int main()
{
    freopen("forest.in","r",stdin);
    freopen("forest.out","w",stdout);

    int test=read(),T;
    n=read(),m=read(),T=read();
    if(test==1||test==2) {Violence::Solve(T); return 0;}
    if((test>=3 && test<=9)||(test>=14 && test<=17)) {Spec::Solve(T);return 0;}
    for(int i=1; i<=n; ++i) LCT::val[i]=read(),LCT::sz[i]=1;
    for(int u,v,i=1; i<=m; ++i) u=read(),v=read(),LCT::Link(u,v);
    int ans=0,x,y,k; char opt[3];
    while(T--)
    {
        scanf("%s",opt), x=read()^ans, y=read()^ans;
        if(opt[0]=='Q') k=read()^ans, LCT::Split(x,y);
        else LCT::Link(x,y);
    }
    return 0;
}

T3

想写树剖没时间了(枚举删边就行啊。。)

#include <cstdio>
#include <cctype>
#define gc() getchar()
typedef long long LL;
const int N=2e5+5;
const LL INF=1e10;

int n,root,Enum,H[N],to[N<<1],nxt[N<<1],val[N<<1];
LL Max;

inline int read()
{
    int now=0,f=1;register char c=gc();
    for(;!isdigit(c);c=gc()) if(c=='-') f=-1;
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now*f;
}
inline void AddEdge(int u,int v,int w)
{
    to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum, val[Enum]=w;
    to[++Enum]=u, nxt[Enum]=H[v], H[v]=Enum, val[Enum]=w;
}
void DFS(int x,int f,LL d)
{
    if(d>Max) Max=d, root=x;
    for(int i=H[x]; i; i=nxt[i])
        if(to[i]!=f) DFS(to[i],x,d+val[i]);
}
namespace Sub
{
    #define lson l,m,rt<<1
    #define rson m+1,r,rt<<1|1

    int dep[N],sz[N],son[N],son_v[N],top[N],cnt,pos[N],fa[N];
    struct Seg_tree
    {
        int sum[N],A[N],tag[N],id[N];
        inline void PushUp(int rt){
            sum[rt]=sum[rt<<1]+sum[rt<<1|1];
        }
        inline void Add(int x,int v){
            sum[x]+=v, tag[x]+=v;
        }
        void Build(int l,int r,int rt)
        {
            if(l==r) sum[rt]=A[l], id[l]=rt;
            else if(l<r)
            {
                int m=l+r>>1;
                Build(lson), Build(rson);
                PushUp(rt);
            }
        }
        void Modify(int l,int r,int rt,int L,int R)
        {
            if(L<=l&&r<=R) Add(rt,1);
            else
            {
                int m=l+r>>1;
                if(L<=m) Modify(lson,L,R);
                if(m<R) Modify(rson,L,R);
                PushUp(rt);
            }
        }
    }t;
    void DFS1(int x)
    {
        sz[x]=1; int mx=0;
        for(int v,i=H[x]; i; i=nxt[i])
            if((v=to[i])!=fa[x])
            {
                fa[v]=x, dep[v]=dep[x]+1, DFS1(v), sz[x]+=sz[v];
                if(mx<sz[v]) mx=sz[v],son[x]=v,son_v[x]=i;
            }
    }
    void DFS2(int x,int tp)
    {
        top[x]=tp;
        if(son[x])
        {
            DFS2(son[x],tp), t.A[pos[son_v[x]]=++cnt]=val[son_v[x]];
            for(int i=H[x]; i; i=nxt[i])
                if(to[i]!=fa[x] && to[i]!=son[x])
                    DFS2(to[i],to[i]);
        }
    }
    void Solve()
    {
        DFS1(1), DFS2(1,1);
        for(int i=1; i<=n; ++i) ;
        
    }
}

int main()
{
    freopen("diameter.in","r",stdin);
    freopen("diameter.out","w",stdout);

    n=read();
    for(int u,v,w,i=1; i<n; ++i) u=read(),v=read(),w=read(),AddEdge(u,v,w);
    Max=-INF, DFS(1,-1,0);
    int u=root,v;
    Max=-INF, DFS(u,-1,0), v=root;
    printf("%I64d\n",Max);
//  Sub::Solve();
    printf("233");

    return 0;
}

转载于:https://www.cnblogs.com/SovietPower/p/8624263.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值