洛谷 P1084 疫情控制 —— 二分+码力

题目：https://www.luogu.org/problemnew/show/P1084

5个月前曾经写过一次，某个上学日的深夜，精疲力竭后只有区区10分，从此没管...

#include<iostream>

#include<cstdio>

#include<cstring>

#include<algorithm>

#include<queue>

#define ll long long

using namespace std;

priority_queue< pair<int,int> >q1,q2;

int const MAXN=;

int n,m,head[MAXN],ct,ans,fa[MAXN],sid[MAXN],lf,lef[MAXN],gr[MAXN];

ll l,r;

bool vis[MAXN],in[MAXN];

struct N{

    int to,next,w;

    N(int t=,int n=,int w=):to(t),next(n),w(w) {}

}edge[MAXN<<];

struct E{

    int st,pos;

    ll rest;

}am[MAXN];

void add(int x,int y,int z)

{

    edge[++ct]=N(y,head[x],z);head[x]=ct;

    edge[++ct]=N(x,head[y],z);head[y]=ct;

}

void ps(int x)

{

    int nw=am[x].st;

    while(fa[nw]!=&&am[x].rest-sid[nw])

    {

        am[x].rest-=sid[nw];

        nw=fa[nw];

    }

    if(fa[nw]==&&am[x].rest>=*sid[nw])am[x].pos=,am[x].rest-=sid[nw];

    else am[x].pos=nw;

}

void init(int x,int f,int g,int w)//fa,gr,lef,sid

{

    int t=;

    fa[x]=f;sid[x]=w;

    if(g==)gr[x]=x;

    else gr[x]=g;

    for(int i=head[x],u;i;i=edge[i].next)

    {

        u=edge[i].to;

        if(u==f)continue;

        init(u,x,gr[x],edge[i].w);t++;

    }

    if(!t)lef[++lf]=x;

}

void dfs(int x)

{

    vis[x]=;

    for(int i=head[x],u;i;i=edge[i].next)

        if(edge[i].to!=fa[x])dfs(edge[i].to);

}

bool pd(ll mid)

{

//    printf("mid=%d\n",mid);

    memset(vis,,sizeof vis);

    memset(in,,sizeof in);

    while(q1.size())q1.pop();

    while(q2.size())q2.pop();

    for(int i=;i<=m;i++)

    {

        am[i].rest=mid;

        ps(i);

        if(am[i].pos!=)dfs(am[i].pos);

        q1.push(make_pair(-am[i].rest,i));

    }

//    for(int i=1;i<=m;i++)

//        printf("am[%d].pos=%d\n",i,am[i].pos);

    for(int i=;i<=lf;i++)

        if(!vis[lef[i]]&&!in[gr[lef[i]]])

        {

            q2.push(make_pair(-sid[gr[lef[i]]],gr[lef[i]]));

            in[gr[lef[i]]]=;

        }

//    printf("q1:%d q2:%d\n",q1.size(),q2.size());

    while(q1.size()&&q2.size())

    {

        int x=q1.top().second;q1.pop();//am

        int y=q2.top().second;q2.pop();//gr

//        printf("x=%d y=%d\n",x,y);

        while(am[x].rest<sid[gr[y]]&&q1.size())x=q1.top().second,q1.pop();

        if(am[x].rest<sid[gr[y]]&&!q1.size())return ;

    }

    if(q2.size())return ;

    else return ;

}

int main()

{

    scanf("%d",&n);

    for(int i=,x,y,z;i<n;i++)

    {

        scanf("%d%d%d",&x,&y,&z);

        add(x,y,z);r+=z;

    }

    init(,,,);

//    for(int i=1;i<=lf;i++)

//        printf("lef[%d]=%d\n",i,lef[i]);

    scanf("%d",&m);

    for(int i=;i<=m;i++)

        scanf("%d",&am[i].st);

    ans=-;

    while(l<=r)

    {

        ll mid=((l+r)>>);

//        cout<<ans<<endl;

        if(pd(mid))ans=mid,r=mid-;

        else l=mid+;

    }

    printf("%d",ans);

    return ;

}

囧1

这几天又重新写它了，自信码力已和当时不可同日而语，于是又写了一遍，然而仅仅20分，改了改成了30分...

#include<iostream>

#include<cstdio>

#include<cstring>

#include<algorithm>

#include<queue>

#include<vector>

using namespace std;

typedef long long ll;

int const xn=;

int n,m,cnt,hd[xn],ct,to[xn<<],nxt[xn<<],w[xn<<],rs[xn],lf[xn],rk[xn];

int f[xn][],son[xn],dep[xn],num[xn];

ll dis[xn],sum,ans;

bool vis[xn],use[xn];

priority_queue<int>q;

vector<int>v[xn];

struct N{

    int st,pos; ll tim;

    bool operator < (const N &y) const

        {return tim<y.tim;}

}a[xn];

int rd()

{

    int ret=,f=; char ch=getchar();

    while(ch<''||ch>''){if(ch=='-')f=; ch=getchar();}

    while(ch>=''&&ch<='')ret=(ret<<)+(ret<<)+ch-'',ch=getchar();

    return f?ret:-ret;

}

void add(int x,int y,int z){to[++ct]=y; nxt[ct]=hd[x]; w[ct]=z; hd[x]=ct;}

int dfs(int x,int fa,int r)

{

    int ret=,fl=; f[x][]=fa;

    for(int i=;i<=;i++)f[x][i]=f[f[x][i-]][i-];

    for(int i=hd[x],u;i;i=nxt[i])

    {

        if((u=to[i])==fa)continue;

        fl=;

        dis[u]=dis[x]+w[i]; dep[u]=dep[x]+;

        if(x==)lf[u]+=dfs(u,x,u),son[++cnt]=u;

        else ret+=dfs(u,x,r);

    }

    if(!fl)ret=,rs[x]=r;

    return ret;

}

bool cmp(int a,int b){return dis[son[a]]>dis[son[b]];}

void dfs2(int x)

{

    bool fl=;

    for(int i=hd[x],u;i;i=nxt[i])

        if((u=to[i])!=f[x][])fl=,dfs2(u);

    if(!fl)num[rs[x]]++;

}

bool ck(int mid)

{

    memset(vis,,sizeof vis);

    memset(num,,sizeof num);

    memset(use,,sizeof use);

    for(int i=;i<=cnt;i++)v[i].clear();

    while(q.size())q.pop();

    for(int i=;i<=m;i++)a[i].pos=;

    for(int i=;i<=m;i++)

    {

        int x=a[i].st; ll ret=mid;

        for(int j=;j>=;j--)

            if(f[x][j]>&&dis[x]-dis[f[x][j]]<=ret)

                ret-=dis[x]-dis[f[x][j]],x=f[x][j];

        if(dep[x]>||(dep[x]==&&ret<dis[x]))dfs2(x);

        else

        {

            a[i].pos=x; a[i].tim=ret-dis[x];

            q.push(i); //x:1~n

            v[x].push_back(i);

//            printf("in:%d\n",i);//tim从大到小

        }

    }

    for(int i=;i<=cnt;i++)

        if(num[son[i]]==lf[son[i]])vis[i]=;//i != son[i]   // vis[1~cnt]

    for(int i=;i<=cnt;i++)

    {

        int x=rk[i];//1~cnt

//        printf("vis[%d]=%d\n",son[x],vis[x]);

        if(vis[x])continue;

        if(!q.size())return ;

        int k=q.top(); q.pop();

        while(use[k])k=q.top(),q.pop();

        if(a[k].tim<dis[son[x]])

        {

            int us=-;

            for(int j=;j<v[son[x]].size();j++)

            {

                int nw=v[son[x]][j];

                if(!use[nw]&&(us==-||a[nw].tim<a[us].tim))us=nw;

            }

            if(us==-)return ;

            else use[us]=,q.push(k);

        }

        vis[x]=;

//        printf("%d -> %d\n",i,son[x]);

    }

//    printf("return 1\n");

    return ;

}

int main()

{

    n=rd();

    for(int i=,x,y,z;i<n;i++)

    {

        x=rd(); y=rd(); z=rd();

        add(x,y,z); add(y,x,z);

        sum+=z;

    }

    m=rd();

    for(int i=;i<=m;i++)a[i].st=rd();

    dfs(,,);

    for(int i=;i<=cnt;i++)rk[i]=i;

    sort(rk+,rk+cnt+,cmp);//rk -> 1~cnt

//    for(int i=1;i<=cnt;i++)printf("dis[%d]=%lld rk[%d]=%d\n",son[i],dis[son[i]],i,rk[i]);

    ll l=,r=sum; ans=-;

    while(l<=r)

    {

        ll mid=((l+r)>>1ll);

//        printf("l=%lld r=%lld mid=%lld\n",l,r,mid);

        if(ck(mid))ans=mid,r=mid-;

        else l=mid+;

    }

    printf("%lld\n",ans);

    return ;

}

囧2

然后又去参考了TJ...看到思路和我的一样，但是代码简洁很多...

于是改改改，然后和模仿的那篇TJ拍拍拍，居然拍出了TJ的错...又和另一篇TJ拍拍拍，大数据又有错...

不管了直接交上去，就A了...

然后发现数据生成没管军队不在根上，但第一篇TJ小数据真的错了...

和 Narh 的代码拍，大数据又有错...但是数据那么大怎么改...不管了(反正也A了)...

又练习码力了，没事不要多写什么 queue 啦， vector 啦，直接排个序就好了。

代码如下：

#include<iostream>

#include<cstdio>

#include<cstring>

#include<algorithm>

using namespace std;

typedef long long ll;

int const xn=;

int n,m,hd[xn],ct,to[xn<<],nxt[xn<<],w[xn<<],dis[xn],sum,ans;

int f[xn][],num[xn],mn[xn],cnta,cnts,st[xn];

bool vis[xn],use[xn];

struct N{

    int id,tim;

    bool operator < (const N &y) const

        {return tim>y.tim;}

}a[xn],son[xn];

int rd()

{

    int ret=,f=; char ch=getchar();

    while(ch<''||ch>''){if(ch=='-')f=; ch=getchar();}

    while(ch>=''&&ch<='')ret=(ret<<)+(ret<<)+ch-'',ch=getchar();

    return f?ret:-ret;

}

void add(int x,int y,int z){to[++ct]=y; nxt[ct]=hd[x]; w[ct]=z; hd[x]=ct;}

void dfs(int x,int fa,int r)

{

    int ret=; f[x][]=fa;

    for(int i=;i<=;i++)f[x][i]=f[f[x][i-]][i-];

    for(int i=hd[x],u;i;i=nxt[i])

    {

        if((u=to[i])==fa)continue;

        dis[u]=dis[x]+w[i];    dfs(u,x,r);

    }

}

bool dfs2(int x)

{

    bool fl=,bj=;

    if(vis[x])return ;

    for(int i=hd[x],u;i;i=nxt[i])

    {

        if((u=to[i])==f[x][])continue; bj=;

        bool k=dfs2(u);

        if(!k)

        {

            fl=;

            if(x==)son[++cnts].id=u,son[cnts].tim=dis[u];

        }

    }

    if(bj)return ;

    return fl;

}

bool ck(int mid)

{

    memset(vis,,sizeof vis);

    memset(use,,sizeof use);

    memset(mn,,sizeof mn);

    cnta=; cnts=;

    for(int i=;i<=m;i++)

    {

        int x=st[i]; ll ret=mid;

        for(int j=;j>=;j--)

            if(f[x][j]>&&dis[x]-dis[f[x][j]]<=ret)

                ret-=dis[x]-dis[f[x][j]],x=f[x][j];

        if(f[x][]>||(f[x][]==&&ret<dis[x]))vis[x]=;

        else

        {

            a[++cnta].id=i; a[cnta].tim=ret-dis[x];//-dis[x]

            if(!mn[x]||a[cnta].tim<a[mn[x]].tim)mn[x]=i;//mn[1~n]=(1~m)

        }

    }

    if(dfs2())return ;

    sort(a+,a+cnta+); sort(son+,son+cnts+);

    for(int i=,p=;i<=cnts;i++)

    {

        int nw=son[i].id;

        if(mn[nw]&&!use[mn[nw]]){use[mn[nw]]=; continue;}//优先用小的,不用考虑其它儿子是因为从大往小用

        while(p<=cnta&&use[a[p].id])p++;

        if(p>cnta||a[p].tim<son[i].tim)return ;

        use[a[p].id]=;

    }

    return ;

}

int main()

{

    n=rd();

    for(int i=,x,y,z;i<n;i++)

    {

        x=rd(); y=rd(); z=rd();

        add(x,y,z); add(y,x,z);

        sum+=z;

    }

    m=rd();

    for(int i=;i<=m;i++)st[i]=rd();

    dfs(,,);

    int l=,r=sum; ans=-;

    while(l<=r)

    {

        int mid=((l+r)>>1ll);

        if(ck(mid))ans=mid,r=mid-;

        else l=mid+;

    }

    printf("%d\n",ans);

    return ;

}