题意:
给你n个点对(x,y)表示第i个人怀疑x和y,问你有多少个点对(x,y)被怀疑的人数总和超过p。
思路:
首先是统计下每个人被怀疑的次数,然后排个序,用两个指针从两头枚举,这样就能O(2*n)的时间统计出有多少个点对的和大于p。然后对于每个输入得点对,如果该点对对应的总和大于p(也就是说我们之前有计算过的),那么我们去判断去掉该重复的点后的和是否还大于p,如果不大于答案-1。
#include<cstdio>
#include<map>
#include<cstring>
#include<algorithm>
using namespace std;
typedef __int64 LL;
const int MAX=3*1e5+5;
map<pair<int,int>,int> c;
int n,p,x[MAX],y[MAX],cnt[MAX],num[MAX];
int main(){
scanf("%d%d",&n,&p);
memset(cnt,0,sizeof(cnt));
for(int i=1;i<=n;i++){
scanf("%d%d",&x[i],&y[i]);
if(x[i]>y[i]) swap(x[i],y[i]);
cnt[x[i]]++,cnt[y[i]]++;
num[x[i]]++,num[y[i]]++;
c[make_pair(x[i],y[i])]++;
}
sort(cnt+1,cnt+1+n);
int j=1;
LL ans=0;
for(int i=n;i>=1;i--){
if(cnt[i]>=p) ans+=(LL)(n-1);
else{
while(cnt[i]+cnt[j]<p&&j<=n) j++;
ans+=(LL)(n-j+1);
if(i>=j) ans--;
//printf("i=%d j=%d\n",i,j);
}
}
//printf("%I64d\n",ans);
ans/=2;
for(int i=1;i<=n;i++){
if(num[x[i]]+num[y[i]]>=p&&num[x[i]]+num[y[i]]-c[make_pair(x[i],y[i])]<p){
ans--;
c[make_pair(x[i],y[i])]=0;
}
}
printf("%I64d\n",ans);
return 0;
}