思路:
用到了分层的思想,网上的题解看不懂,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;
}