洛谷3505 [POI2010]TEL-Teleportation（思维题）

题意

给一张图，要求你再尽可能的多连边，使得从1到2至少要经过5条边。

题解

分层+方案统计
至少要经过5条边，不妨设1所在的块为A，接下来有B，C，D，E，F，其中2在第F块。
一个块是距离1一定距离的集合，所以块内怎么连边都无所谓（甚至连成一个完全图）。但块外不能从B连到D，否则从A到F的最短距离变成了4。
In a word，一个点只允许和块内或相邻块的点连边。

再来考虑一下每个点分到那个块的问题。
首先，1、2绝对是独立地占有A和F。
跟1距离为1，放在B，与2相连，也就是的跟2距离为1，放在E，这也是板上钉钉的事。
剩下的问题就是把自由的点分配到C和D中。若一个点有与B或E连边的，那么它所处的块也固定了。
再剩下的，是完全自由的，那么那里方案多就往那放（显然要么全放C，要么全放D）。

方案计数分1、2连出的边，每一块中互相连边，相邻两块的边来考虑，基本的数学素养，不懂自己看代码吧(─.─|||

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=40010,MAXM=1000010;

int n,m;
int bel[MAXN];

struct E{int y,next;}e[MAXM*2];int len=1,last[MAXN];
void ins(int x,int y)
{
    e[++len]=(E){y,last[x]};last[x]=len;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int x,y;scanf("%d%d",&x,&y);
        ins(x,y);ins(y,x);
    }
    
    int s1=0,s2=0,s3,ans=0,t1=0,t2=0;
    for(int k=last[1];k;k=e[k].next) bel[e[k].y]=1,s1++;
    for(int k=last[2];k;k=e[k].next) bel[e[k].y]=2,s2++;
    s3=n-s1-s2-2;
    ans=s1+s2+s1*(s1-1)/2+s2*(s2-1)/2+s3*(s3-1)/2;
    for(int i=3;i<=n;i++) if(!bel[i])//debug 要写的，有些已经被1/2更新过的点不计入 
        for(int k=last[i];k;k=e[k].next) if(bel[e[k].y])//debug bel[i].y 
        {
            if(bel[e[k].y]==1) t1++;//debug bel[i].y 
            else t2++;
            break;
        }
    ans+=t1*s1+t2*s2+(s3-t1-t2)*max(s1,s2);
    printf("%d\n",ans-m);//debug ans-m
    return 0;
}