LeetCode 5683. 统计点对的数目（容斥）

题意：

给你一个无向图，无向图由整数 n  ，表示图中节点的数目，和 edges 组成，
其中 edges[i] = [ui, vi] 表示 ui 和 vi 之间有一条无向边。同时给你一个代表查询的整数数组 queries 。

第 j 个查询的答案是满足如下条件的点对 (a, b) 的数目：

a < b
cnt 是与 a 或者 b 相连的边的数目，且 cnt 严格大于 queries[j] 。
请你返回一个数组 answers ，其中 answers.length == queries.length 且 answers[j] 是第 j 个查询的答案。

请注意，图中可能会有 重复边 。

数据范围:
2 <= n <= 2 * 1e4
1 <= edges.length <= 1e5
1 <= ui, vi <= n
ui != vi
1 <= queries.length <= 20
0 <= queries[j] < edges.length

解法：

设d[i]为点i的度数.

这题的难点主要在于满足条觉得点a和点b可能是相连的,
那么d[a]+d[b]需要减去边(a,b),有可能减完之后就满足条件了.

如果不用考虑两点相连,那么做法是对d[]排序,然后双指针计算合法的数据组数.

考虑容斥,按上述方法计算完答案之后,容斥掉不合法的方案.
设查询的数为p,那么以下情况是不合法的:
1.d[i]+d[i]>p,即自己和自己匹配.
2.a和b相连,设mp(a,b)是a和b的公共边数量,
如果d[a]+d[b]>p,但d[a]+d[b]-mp(a,b)<=p,即错误的统计了公共边.

code：

#define PI pair<int,int>
const int maxm=1e5+5;
class Solution {
public:
    vector<int> countPairs(int n, vector<vector<int>>& e, vector<int>& q) {
        vector<int>ans;
        //建图
        vector<int>d(n);
        map<PI,int>mp;
        for(auto i:e){
            int x=i[0]-1,y=i[1]-1;
            if(x>y)swap(x,y);
            mp[{x,y}]++;
            d[x]++,d[y]++;
        }
        vector<int>temp=d;
        sort(temp.begin(),temp.end());
        for(int p:q){
            int sum=0;
            //双指针
            int j=n-1;
            for(int i=0;i<n;i++){
                while(j>=0&&temp[i]+temp[j]>p){
                    j--;
                }
                sum+=(n-1)-j;
            }
            //自己和自己匹配
            for(int i=0;i<n;i++){
                if(temp[i]+temp[i]>p)sum--;
            }
            //相邻匹配
            for(auto [xx,yy]:mp){
                int x=xx.first,y=xx.second;
                int t=d[x]+d[y];
                if(t>p&&t-yy<=p)sum-=2;
            }
            sum/=2;
            ans.push_back(sum);
        }
        return ans;
    }
};

---