Codeforces 1634 E. Fair Share —— dfs

This way

题意：

给你m个数组，你要将每个数组的数字对半分到多重集合L和R中，并且最终L和R相同，问你对于每个数组的每个数字怎么分。

题解：

其实还是比较简单的题目…首先假设我们拿出了第一行第一列的数x放到左边，那么肯定要在所有数组中找到另一个x放到右边。
如果这两个x在同一行那么就相当于这行就多了一组左右。
如果不在同一行，假设L拿到的x是在行i，R拿到的x是在行j。那就没有办法保证每一行都是对半分啊？那我们在j行随便拎一个数y到L里面呗，再继续往下找另一个y放到R中。如此往复R一定能先回到i。所以这里其实不用特判每一行是否L有多取。
如果在过程中找不到一对相同的数，那肯定是无解了。

所以我在这里枚举每个数组然后随便挑一个数加到L中，然后对于这个数对应的找到另一个相同的数加到R中。所以lef[i]表示第i行，里面存的是每个数的值和列号。rig[i]表示对于值为i的数出现的行号列号。

#include<bits/stdc++.h>
using namespace std;
#define pii pair<int,int>
const int N=2e5+5;
vector<int>ans[N];
map<int,multiset<pii> >lef,rig;
int f=0,num[N];
void dfs(int x){
    if(lef[x].empty()||num[x]>=ans[x][0]/2)
        return ;
    pii v=*lef[x].begin();
    ans[x][v.second]=0;
    num[x]++;
    lef[x].erase(lef[x].begin());
    if(rig[v.first].find({x,v.second})==rig[v.first].end()){
        f=1;
        return ;
    }
    rig[v.first].erase(rig[v.first].find({x,v.second}));
    if(rig[v.first].size()==0){
        f=1;
        return ;
    }
    pii u=*rig[v.first].begin();
    ans[u.first][u.second]=1;
    rig[v.first].erase(rig[v.first].begin());
    if(lef[u.first].find({v.first,u.second})==lef[u.first].end()){
        f=1;
        return ;
    }
    lef[u.first].erase(lef[u.first].find({v.first,u.second}));
    dfs(u.first);
}
int main()
{
    int m;
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        int n,x;
        scanf("%d",&n);
        ans[i]=vector<int>(n+5,0);
        ans[i][0]=n;
        for(int j=1;j<=n;j++){
            scanf("%d",&x);
            lef[i].insert({x,j});
            rig[x].insert({i,j});
        }
    }
    for(int i=1;i<=m;i++){
        while(!f&&num[i]<ans[i][0]/2&&lef[i].size())
            dfs(i);
    }
    if(f)return 0*printf("NO\n");
    printf("YES\n");
    for(int i=1;i<=m;i++){
        for(int j=1;j<=ans[i][0];j++)
            printf("%s",ans[i][j]?"R":"L");
        printf("\n");
    }
    return 0;
}