BZOJ 2088/POI 2010 Teleportation

思路：
用到了分层的思想，网上的题解看不懂，orz hta 的方法：
首先我们把1当成源点 2当初汇点（并不是网络流 只是这样写起来方便）
然后将与1直接相连的点设成第一层（集合s1） 与2直接相连的点设置成倒数第一层(集合s2)
然后剩下的点放在中间(s3)

首先由于题目中给的初始状态是合法的 所以中间的点不会同时和s1,s2连边 不然最短路变成4
那么我们现在来考虑每层之间的连边情况
显然是可以两两连边的 这样的话答案不会不合法（不合法当且尽当一条边跨集合才可能出现缩短最短路）

在来考虑跨集合的情况：
对于一个在s3中的点 如果本来就和s1连边 则不能再和s2中的点连边
这样我们索性将它与s1所有没有连过的点连边 反之亦然
那对于一个孤立的点 连|s1|，|s2|中较大的那个答案肯定更优；

这样连边的话为什么是对的呢？
正确性：
不会出现有一条最短路len<=4首先1->s1->s3->s2->2要四条边
在s3中至少还得经过一条边 所以显然成立

最优性：
暂时难以证明。。。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<queue>
using namespace std;
typedef long long LL;
const int imax=40000+229;
const int dmax=imax;
const int bmax=1000000+229; 
int n,m,x[bmax],y[bmax];
LL s1,s2,s3;
int f1[imax],f2[imax],Link[imax];
int l[imax],r[imax];

void iread()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++) 
    {
        scanf("%d%d",&x[i],&y[i]);  
        if(x[i]>y[i]) swap(x[i],y[i]);
        if(x[i]==1) s1++,Link[y[i]]=1;
        else if(x[i]==2) s2++,Link[y[i]]=2;
    }
}

void iwork()
{
    s3=n-s1-s2-2;
    LL ans=((s1+1)*(s1)/2)+((s2+1)*s2/2)+((s3-1)*s3/2);

    for(int i=1;i<=n;i++) f1[i]=s1,f2[i]=s2;
    for(int i=1;i<=m;i++) 
        {
            if(Link[x[i]]==1) l[y[i]]=1,f1[y[i]]--;
            if(Link[y[i]]==1) l[x[i]]=1,f1[x[i]]--;
            if(Link[x[i]]==2) r[y[i]]=1,f2[y[i]]--;
            if(Link[y[i]]==2) r[x[i]]=1,f2[x[i]]--;
        }

    for(int i=1;i<=n;i++)
        if(i!=1 && i!=2 && !Link[i])
        { 
            if(!l[i] && !r[i]) ans+=max(s1,s2);
            else if(r[i]) ans+=s2;
            else ans+=s1; 
        }
    printf("%lld\n",ans-m);
}

int main()
{
//  freopen("p3.in","r",stdin);
    iread();
    iwork();
    return 0;
}