[CF538H]Summer Dichotomy

[CF538H]Summer Dichotomy

题目大意：

​ 将若干个学生分为两个班级\(S_1,S_2\)，每个班的学生数分别为\(n_1,n_2\)（甚至可以没有学生，也可以没有老师）。给出限制\(t_{\min},t_{\max}\)，要求\(t_{\min}\le n_1+n_2\le t_{\max}\)。有\(n(n\le10^5)\)个老师，每个老师希望他所任教的班级人数在\([l_i,r_i]\)范围内。有\(m(m\le10^5)\)对老师之间有一些私人恩怨，不能分在一个班级。问是否存在合法的分班方案。如果有，求出其中的任意一种，输出每个班的总人数以及各个老师所任教的班级。

思路：

​ 对于所有\([l_i,r_i]\)的限制，我们不妨假设\(n_1=\min\{r_i\},n_2=\max\{l_i\}\)，显然这是比较松的约束。再考虑\(t_{\min},t_{\max}\)的限制，确定可行的一组\(n_1,n_2\)。考虑二分图染色构造老师分配的方案。对于只能分到\(S_1\)或只能分到\(S_2\)的老师DFS遍历染色，若更新到的结点与已染色结点矛盾，说明根本不是二分图，不存在合法的方案。对于两个都不可以分进去的，说明也不存在合法方案。最后再对于\(S_1\)和\(S_2\)都可以的进行染色。

源代码：

#include<cstdio>
#include<cctype>
#include<vector>
#include<climits>
inline int getint() {
    register char ch;
    while(!isdigit(ch=getchar()));
    register int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
constexpr int N=1e5;
int l[N],r[N],ans[N];
std::vector<int> e[N];
inline void add_edge(const int &u,const int &v) {
    e[u].push_back(v);
    e[v].push_back(u);
}
void dfs(const int &x,const int &c) {
    if(ans[x]) {
        if(ans[x]!=c) throw(0);
        return;
    }
    ans[x]=c;
    for(auto &y:e[x]) {
        dfs(y,3-c);
    }
}
int main() {
    const int t_min=getint(),t_max=getint();
    const int n=getint(),m=getint();
    int n1=INT_MAX,n2=INT_MIN;
    for(register int i=0;i<n;i++) {
        n2=std::max(n2,l[i]=getint());
        n1=std::min(n1,r[i]=getint());
    }
    if(n1+n2<t_min) n2=t_min-n1;
    else if(n1+n2>t_max) n1=t_max-n2;
    if(n1<0||n2<0) {
        puts("IMPOSSIBLE");
        return 0;
    }
    for(register int i=0;i<m;i++) {
        add_edge(getint()-1,getint()-1);
    }
    for(register int i=0;i<n;i++) {
        try {
            if(!(l[i]<=n1&&n1<=r[i])&&!(l[i]<=n2&&n2<=r[i])) throw(0);
            if((l[i]<=n1&&n1<=r[i])&&!(l[i]<=n2&&n2<=r[i])) dfs(i,1);
            if((l[i]<=n2&&n2<=r[i])&&!(l[i]<=n1&&n1<=r[i])) dfs(i,2);
        } catch(...) {
            puts("IMPOSSIBLE");
            return 0;
        }
    }
    for(register int i=0;i<n;i++) {
        try {
            if(!ans[i]) dfs(i,1);
        } catch(...) {
            puts("IMPOSSIBLE");
            return 0;
        }
    }
    puts("POSSIBLE");
    printf("%d %d\n",n1,n2);
    for(register int i=0;i<n;i++) {
        printf("%d",ans[i]);
    }
    return 0;
}