ZOJ - 3780 Paint the Grid Again 拓扑排序+思维

ZOJ - 3780 Paint the Grid Again

题意：

一个 n*n 的方格，有两种操作： Ri 将第 i 行 全部全部涂成黑色，Cj将第 j 列全部涂成白色（X 为黑色，O为白色）。一开始方格没有任何颜色（不是白色，也不是黑色），每一次操作都会覆盖方格内的颜色。给定最后矩阵的颜色，求最少需要多少次操作才能将原矩阵变为给定的矩阵。输出每一次操作。（如果可以同时进行R 和 C 操作的时候先进行R操作）

题解：

问题的关键点就是在于怎么建边，只要边建出来了，那问题就基本上解决了。

我们把每一种操作抽象成为一个点，比如R1操作，就是把第一行全涂成X 把这个操作抽象为一个点，标号为1。这样的操作一共有多少个呢？R1 R2 R3…Rn, C1,C2,C3…Cn。一共有2n个操作，所以就有2n个点。标号怎么给呢？因为R操作的优先级高于C  所以我们把R操作标记为 1~n，C操作标记为 n+1~2n;。

点解决了，那边呢？？

我们考虑每一个格子，如果这个格子(i,j)里是X,那么这个格子的最终操作一定是Ri 所以说，Ri操作一定会在C（j+n）这个操作之后，所以我们建边是 (j+n)->i,同样如果这个格子是O，那么这个格子的最终操作一定是C(j+n),所以Ri一定在C（j+n）之前 所以我们建边 i->(j+n);

in数组存的是这个点的入度，拓扑排序时需要用到。d数组存的从这个点出发的边有多少条。如果从这个点出发的边有n条的话，那么这个操作就是非必须的操作，在答案中就不用存它了（1~n中的某一个点与 n+1~2n中的点全部相连 或 n+1~2n中的某一点与1~n的全部点相连。例如：先将第 y （y属于 n+1~2n）列涂成白色，与 y 相连的有 n 条边，这 n 条边的另一个端点属于 1~n（行操作R ），最后第 y 列会全部变为黑色，所以一开时涂与不涂最后的结果都一样，因为要求操作数最少，所以忽略 ）。
还有一个细节是，拓扑排序的时候要用优先队列，因为输出顺序有优先级嘛~

参考博客：https://blog.csdn.net/weixin_43179892/article/details/89449301

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<sstream>
#include<cstring>
#include<bitset>
#include<cstdio>
#include<string>
#include<deque>
#include<stack>
#include<cmath>
#include<queue>
#include<set>
#include<map>
#define mod 1000000007
using namespace std;
typedef long long ll;
const int maxn = 3e5+10;
struct node
{
    int v;
    int next;
}e[2*maxn];
int head[maxn];
int in[maxn]; //入度
int d[maxn];   //以该点为起点的边有多少条
char mp[510][510];
int ans[maxn];
int n,cnt,tot;

void init()
{
    cnt = tot = 0;
    memset(d,0,sizeof(d));
    memset(in,0,sizeof(in));
    memset(head,-1,sizeof(head));

}
void add(int u,int v)
{
    e[cnt].v = v;
    e[cnt].next = head[u];
    head[u] = cnt++;
}

int tuopu()
{
    priority_queue<int, vector<int>, greater<int> > q;  //小根堆
    for(int i=1;i<=2*n;i++)
    {
        if(in[i]==0)
            q.push(i);
    }

    while(!q.empty())
    {
        int u = q.top();
        q.pop();
       if(d[u]!=n)   // 1~n中的某一个点与 n+1~2n中的点全部相连 或 n+1~2n中的某一点与1~n的全部点相连
            ans[tot++] = u;
        for(int i=head[u] ; i!=-1 ; i=e[i].next)
        {
            int v = e[i].v;
            in[v]--;
            if(in[v]==0)
                q.push(v);
        }
    }
    int flag = 1;
    for(int i=1;i<=2*n;i++)
    {
        if(in[i])
        {
            flag = 0;
            break;
        }
    }
    return flag;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        init();
        scanf("%d ",&n);
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                scanf("%c",&mp[i][j]);
                if(mp[i][j]=='O')    //建边
                {
                    add(i,j+n);
                    d[i]++;
                    in[j+n]++;
                }
                else
                {
                    add(j+n,i);
                    d[j+n]++;
                    in[i]++;
                }
            }
            getchar();
        }

        if( !tuopu() )
            printf("No solution\n");
        else
        {
            for(int i=0 ; i<tot ; i++)
            {
                if(ans[i]>n)
                    printf("C%d",ans[i]-n);
                else
                    printf("R%d",ans[i]);
                if(i==tot-1)
                    printf("\n");
                else
                    printf(" ");
            }
        }
    }
    return 0;
}