直接硬着模拟就好了,因为 k ∗ n < = 500000 k*n<=500000 k∗n<=500000,所以可以把这 k k k张图目前的拓扑度存下来
每次遇到一个节点 x x x,我们知道在哪些图中这个节点的入读为 0 0 0,就输出那张图的编号即可
选哪个图都无所谓
#include <bits/stdc++.h>
using namespace std;
const int maxn = 5e5+10;
int n,k,m,id[maxn];
vector<int>vec[maxn],in[maxn],avi[maxn];
int main()
{
cin >> n >> k >> m;
for(int i=1;i<=k;i++)
for(int j=1;j<=n;j++)
in[i].push_back( 0 );
for(int i=1;i<=m;i++)
{
int l,r; cin >> l >> r;
vec[l].push_back( r );
for(int j=1;j<=k;j++) in[j][r-1]++;
}
for(int i=1;i<=k;i++)
for(int j=1;j<=n;j++)
{
if( in[i][j-1]==0 ) avi[j].push_back( i );
}
for(int i=1;i<=n*k;i++)
{
int x; cin >> x;
int z = avi[x][id[x]++];
cout << z << " ";
for(auto v:vec[x] )
if( --in[z][v-1]==0 ) avi[v].push_back( z );
}
}