题意就是说,每个人就2个投票权,当然他们都不会投给自己,然后让你选择2个人员id使得,两个id加起来的票数大于等于p。当然如果你选择的id是:1 2 , 而投票的人中存在同时投给1和2的人,那么也只能算1票。
思路:先是什么都不管,先枚举第一位1~n,第二位通过二分来找到可行的数量,当然其中某些可行的方案是伪的,这些伪方案中讲上述举的例子所说的(你选择的id是:1 2 , 而投票的人中存在同时投给1和2的人)这里算成了2票,那么先不管,全部算起来,之后再处理,这里要注意算出来的值要除以2,因为你选1 2和选2 1算相同方案。
既然存在这种伪方案,并且我可以肯定这些伪方案必定在1~n个投票的人中的几种。所以我干脆枚举方案,然后判断他是否成为了伪方案,如果是,那么答案减1即可。
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
struct DATA
{
int x, y;
}dt[300005];
int ct[300005], st[300005];
int n, p;
int fd(int x)
{
if(st[n]<x) return n+1;
int l=1, r=n, mid, ret;
while(l<=r){
mid=(l+r)>>1;
if(st[mid]>=x){
ret=mid;
r=mid-1;
}else l=mid+1;
}
return ret;
}
bool cmp(DATA a,DATA b)
{
if(a.x==b.x) return a.y<b.y;
return a.x<b.x;
}
bool ok(int i,int j,int num)
{
int tp=ct[i]+ct[j];
int zp=tp-num;
if(tp>=p&&zp<p){
return 1;
}
return 0;
}
int main()
{
int i, j, x, y;
scanf("%d%d",&n,&p);
memset(ct,0,sizeof(ct));
for(i=1;i<=n;i++){
scanf("%d%d",&x,&y);
if(x>y) swap(x,y);
ct[x]++, ct[y]++;
dt[i].x=x, dt[i].y=y;
}
for(i=1;i<=n;i++)
st[i]=ct[i];
sort(st+1,st+1+n);
__int64 ans=0, tp;
for(i=1;i<=n;i++){
if(ct[i]>=p){
ans+=n-1;
}else{
tp=n+1-fd(p-ct[i]);
if(ct[i]>=p-ct[i]) tp--;
ans+=tp;
}
}
ans/=2;
sort(dt+1,dt+1+n,cmp);
int pre;
x=dt[1].x, y=dt[1].y, pre=1;
for(i=2;i<=n;i++){
if(dt[i].x==x&&dt[i].y==y){
pre++;
if(i==n){
if(ok(x,y,pre)){
ans-=1;
}
}
}
if(dt[i].x!=x||dt[i].y!=y){
if(ok(x,y,pre)){
ans-=1;
}
x=dt[i].x, y=dt[i].y, pre=1;
if(i==n){
if(ok(x,y,pre)){
ans-=1;
}
}
}
}
printf("%I64d\n",ans);
return 0;
}