bzoj2088 [Poi2010]Teleportation 分层图

Description

---

Zy大帝拥有n个星球，因为距离非常遥远，所以Zy在他所居住的1号星球和他的军事基地霸中所在的2号星球建造了两个传送门，这样从1号星球到2号星球就只需要250分钟，回去也一样（双向）。由于科技的发展，各个星球陆陆续续建造了和自己居民最经常去的星球之间的传送门，并且他们的传送门只需要1个小时（真快啊！），他们发现和别的星球建设传送门对促进经济发展有很大的帮助，于是向和其他所有星球建设传送门发展，Zy突然发现两两星球的传送门的建设会威胁到他的安全，可是他又想促进自己帝国的发展，于是他请到了他精心培养的你，希望你能帮他解决这个难题。

2<=n<=40000
0<=m<=1000000

输出：还能让多少对传送门建造，但又不会比Zy从1号星球到2号星球快（就是说增加传送门后，从1号星球到2号星球还是Zy的传送门最快）

Solution

---

其实最后一行才是真·题意
注意到1小时=60分钟，而60分钟*5>250分钟，我们可以把图分层，钦定1作为第1层，2作为第5层，与1相邻的点都是第2层，与2相邻的点都是第4层，剩余的点都是第3层，则不能存在一条连通第一层到第五层的路径

每一层内的点两两可以连边
若一个第3层的点连向了第2层，就把它和第2层的所有点连边
若一个第3层的点连向了第4层，就把它和第4层的所有点连边
若一个第3层的点没有连边，那么可以向点数多的相邻层连满边
可以用反证法证明这样建图一定是最优的

我太菜了并没有想出正解

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

const int N=2000005;
const int E=4000005;

struct edge {int x,y,next;} e[E];

int bel[N];
int ls[N],edCnt;

void add_edge(int x,int y) {
    e[++edCnt]=(edge) {x,y,ls[x]}; ls[x]=edCnt;
    e[++edCnt]=(edge) {y,x,ls[y]}; ls[y]=edCnt;
}

int main(void) {
    int n,m; scanf("%d%d",&n,&m);
    rep(i,1,m) {
        int x,y; scanf("%d%d",&x,&y);
        add_edge(x,y);
    }
    int sum2=0,sum3=0,sum4=0;
    for (int i=ls[1];i;i=e[i].next) bel[e[i].y]=2,sum2++;
    for (int i=ls[2];i;i=e[i].next) bel[e[i].y]=4,sum4++;
    sum3=n-2-sum2-sum4;
    int now2=sum2,now4=sum4;
    int ans=0;
    rep(x,3,n) {
        if (!bel[x]) {
            ans+=--sum3;
            int rec=0,cnt=0;
            for (int i=ls[x];i;i=e[i].next) {
                if (bel[e[i].y]) rec=bel[e[i].y],cnt++;
                else if (e[i].y>x) ans--;
            }
            if (rec==2) ans+=sum2-cnt;
            else if (rec==4) ans+=sum4-cnt;
            else ans+=std:: max(sum2,sum4);
        } else {
            for (int i=ls[x];i;i=e[i].next) if (bel[e[i].y]==bel[x]&&e[i].y>x) ans--;
            if (bel[x]==2) ans+=--now2;
            else if (bel[x]==4) ans+=--now4;
        }
    }
    printf("%d\n", ans);
    return 0;
}