bzoj 2088: [Poi2010]Teleportation 分层图

题意

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

分析

我太弱了想不到正解。。。

这题用到了分层图的思想。
首先我们将1设为第一层，2设为第5层，和1直接相连的点设为第二层，和2直接相连的点设为第4层，其余设为第3层。
显然不能有第一层到第5层的路径。
最优的方案显然是每层之间的点可以两两相互连边，若一个第三层的点连到某个第一层的点，则该点可以向第一层的所有点连边。若一个第三层的点连到某个第四层的点，则该点可以向所有第四层的点连边。否则的话就向节点数较多的那一层连边就好了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=40005;

int n,m,cnt,last[N],op[N];
struct edge{int to,next;}e[2000005];

void addedge(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt;
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        addedge(x,y);
    }
    int s1=0,s2=0;
    for (int i=last[1];i;i=e[i].next) op[e[i].to]=1,s1++;
    for (int i=last[2];i;i=e[i].next) op[e[i].to]=2,s2++;
    int ans=0,s3=n-s1-s2-2,now1=s1,now2=s2;
    for (int i=3;i<=n;i++)
        if (!op[i])
        {
            s3--;ans+=s3;
            int flag=0,s=0;
            for (int j=last[i];j;j=e[j].next)
                if (op[e[j].to]) flag=op[e[j].to],s++;
                else if (e[j].to>i) ans--;
            if (flag==1) ans+=s1-s;
            else if (flag==2) ans+=s2-s;
            else ans+=max(s1,s2);
        }
        else
        {
            if (op[i]==1) now1--,ans+=now1;
            else now2--,ans+=now2;
            for (int j=last[i];j;j=e[j].next)
                if (op[e[j].to]&&e[j].to>i) ans--;
        }
    printf("%d",ans);
    return 0;
}