Codeforces 1237C2. Balanced Removals (Harder)

传送门

先来考虑一下二维时的情况，那么对于 $x$ 相同的点，我们按 $y$ 排序，然后相邻的一对对消除

最后 $x$ 坐标相同的点最多剩下一个，那么此时所有点的 $x$ 坐标都不一样

再按 $x$ 把 $x$ 相邻的一对对删除即可

扩展到三维，显然也可以同样的思路，先把 $x,y$ 相同的点按 $z$ 一对对消除，然后在把 $x$ 相同的点按 $y,z$ 相邻的一对对消除

最后按 $x,y,z$ 相邻的一对对消除即可

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
inline int read()
{
    int x=,f=; char ch=getchar();
    while(ch<''||ch>'') { if(ch=='-') f=-; ch=getchar(); }
    while(ch>=''&&ch<='') { x=(x<<)+(x<<)+(ch^); ch=getchar(); }
    return x*f;
}
const int N=5e4+;
int n;
struct dat {
    int x,y,z,id;
    inline bool operator < (const dat &tmp) const {
        if(x!=tmp.x) return x<tmp.x;
        return y!=tmp.y ? y<tmp.y : z<tmp.z;
    }
}A[N];
bool vis[N];
int main()
{
    n=read();
    for(int i=;i<=n;i++)
        A[i].x=read(),A[i].y=read(),A[i].z=read(),A[i].id=i;
    sort(A+,A+n+);
    int l=;
    for(int i=;i<=n;i++)
        if(l && A[i].x==A[l].x && A[i].y==A[l].y)
        {
            vis[i]=vis[l]=;
            printf("%d %d\n",A[i].id,A[l].id);
            l=;
        }
        else l=i;
    l=; while(vis[l]) l++;
    for(int i=l+;i<=n;i++)
    {
        if(vis[i]) continue;
        if(l && A[i].x==A[l].x)
        {
            vis[i]=vis[l]=;
            printf("%d %d\n",A[i].id,A[l].id);
            l=;
        }
        else l=i;
    }
    l=; while(vis[l]) l++;
    for(int i=l+;i<=n;i++)
    {
        if(vis[i]) continue;
        if(l) printf("%d %d\n",A[i].id,A[l].id),l=;
        else l=i;
    }
    return ;
}