CS 400 Digit Permutation 思维+拓扑序

题意:给出n行,第i行有m个数字 每个数字范围为0~k-1.
求出一个[0..k-1]的排列p,使得将n行中,数字x与p[x]替换后满足以下条件.
每行都没有前导0,从1行往下,每行的字典序递增.
k,n*m<=1e5. 无解输出-1,否则输出满足条件的排列p 


第i行字典序要大于第i-1行 则从左往右比较第一个不相同得数字分别为d1,d2 (d1!=d2) ,显然p(d1)<p(d2) （p为排列 不能取等号）.
现在得到n-1个关于p得限制 若这n-1条限制都满足 则显然得到一解.
若p(d1)<p(d2) 则d1->d2连接一条边 当前度为0得赋当前最小值,若有环则无解.

(注意不能有前导0 一开始选度为0并且不在开头得赋值为0即可.)

#include <bits/stdc++.h>using namespace std;typedef long long ll;const int N=1e6+5,mod=1e9+7;vector<int> a[N],e[N];int n,m,k,x,p[N],ban[N],deg[N];queue<int> q;void Topo(){for(int i=0;i<k;i++)if(deg[i]==0&&ban[i]==0)q.push(i); if(q.size()==0){puts("-1");return;}for(int i=0;i<k;i++){if(deg[i]==0&&ban[i]!=0)q.push(i);}int cur=0;while(!q.empty()){int u=q.front();q.pop();p[u]=cur,cur++;for(int i=0;i<e[u].size();i++){int v=e[u][i];deg[v]--;if(deg[v]==0)q.push(v);}}if(cur!=k)printf("-1");elsefor(int i=0;i<k;i++)printf("%d ",p[i]);}int main(){memset(deg,0,sizeof(deg));memset(ban,0,sizeof(ban));cin>>n>>k>>m;for(int i=0;i<n;i++)for(int j=0;j<m;j++)scanf("%d",&x),a[i].push_back(x),ban[a[i][0]]=1;bool flag=true;for(int i=1;i<n;i++){int u,v=-1;for(int j=0;j<m;j++){if(a[i][j]!=a[i-1][j]){u=a[i][j];v=a[i-1][j];break;}}if(v==-1)flag=false;else{deg[u]++;e[v].push_back(u);}}if(flag==false)puts("-1");elseTopo();return 0;}