【HEOI2013】bzoj3168 钙铁锌硒维生素

本文主要是介绍【HEOI2013】bzoj3168 钙铁锌硒维生素，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

Description

银河队选手名单出来了！小林，作为特聘的营养师，将负责银河队选手参加
宇宙比赛的饮食。众所周知，前往宇宙的某个星球，通常要花费好长好长的时间，人体情况在这之间会发生变化，因此，需要根据每天的情况搭配伙食，来保证营养。小林把人体需要的营养分成了n种，这些营养包括但不限于铁，钙。他准备了2套厨师机器人，一套厨师机器人有n个，每个厨师机器人只会做一道菜，这道菜一斤能提供第i种营养xi微克。想要吃这道菜的时候，只要输入一个数，就能吃到对应数量的这道菜了。为防止摄入过量对身体造成的伤害，每个机器人还有防过量摄入药，只要输入一个数，就能生成一定剂量的药，吃了这些药，就能减少相当于食用对应数目的这道菜提供的营养。
小林之所以准备2套厨师机器人，正是因为旅途漫漫，难以预计，也许某一个厨师机器人在途中坏掉，要是影响了银河队选手的身体，就不好了。因此，第2套厨师机器人被用来做第1套的备用。小林需要为每一个第1套厨师机器人选一个第2套厨师机器人作备份，使得当这个机器人坏掉时，用备份顶替，整套厨师机器人仍然能搭配出任何营养需求，而且，每个第2套厨师机器人只能当一个第1套厨师机器人的备份。
Input

第一行包含一个正整数n。接下来n行，每行n个整数，表示第
1套厨师机器人做的菜每一斤提供的每种营养。再接下来n行，每行n个整数，表示第2套厨师机器人做的菜每一斤提供的每种营养。 Output

第一行是一个字符串，如果无法完成任务，输出“NIE”，否则输
出“TAK”，并跟着n行，第i行表示第i个第1套机器人的备份是哪一个第2套机器人。为了避免麻烦，如果有多种可能的答案，请给出字典序最小的那一组。

对于每一个 $b_i$ 求出他能替换哪些向量 $a_j$ ，然后就变成了字典序最小的二分图完备匹配问题。
把向量 $a$ 组成矩阵 $A$ ，然后求 $xA=b_i$ ，也就是 $b_i=x_1a_1+x_2a_2+\cdots +x_na_n$ 。如果 $x_j$ 不为零， $b_i$ 可以替换 $a_j$ 。把 $b$ 拼成矩阵 $B$ ，求出 $C=B*A^{-1}$ 就是二分图的邻接矩阵。
在二分图上先求出一个完备匹配，然后从小到大考虑每个位置连的边，选择一条最小的而且不会影响之前的匹配的路径进行更换。
原题数据貌似不保证矩阵 A <script type="math/tex" id="MathJax-Element-1193">A</script>可逆，需要判断。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define LL long long
const int p=100000007;
int n,f[310],vis[310],ans[310],ok=1;
int pow(int base,int x)
{int ret=1;while (x){if (x&1) ret=(LL)ret*base%p;base=(LL)base*base%p;x>>=1;}return ret;
}
struct mat
{int a[310][310];void rd(){for (int i=1;i<=n;i++)for (int j=1;j<=n;j++)scanf("%d",&a[i][j]);}mat operator * (const mat &m) const{mat ret;for (int i=1;i<=n;i++)for (int j=1;j<=n;j++){ret.a[i][j]=0;for (int k=1;k<=n;k++)ret.a[i][j]=(ret.a[i][j]+(LL)a[i][k]*m.a[k][j]%p)%p;}return ret;}mat inv(){mat ret;for (int i=1;i<=n;i++)for (int j=1;j<=n;j++)ret.a[i][j]=(i==j);for (int i=1;i<=n;i++){int x;x=-1;for (int j=i;j<=n;j++)if (a[j][i]){x=j;break;}if (x==-1){ok=0;return ret;}if (x!=i)for (int j=1;j<=n;j++){swap(a[i][j],a[x][j]);swap(ret.a[i][j],ret.a[x][j]);}x=pow(a[i][i],p-2);for (int j=1;j<=n;j++){a[i][j]=(LL)a[i][j]*x%p;ret.a[i][j]=(LL)ret.a[i][j]*x%p;}for (int j=1;j<=n;j++)if (j!=i){x=a[j][i];for (int k=1;k<=n;k++){a[j][k]=(a[j][k]-(LL)a[i][k]*x%p+p)%p;ret.a[j][k]=(ret.a[j][k]-(LL)ret.a[i][k]*x%p+p)%p;}}}return ret;}
}a,b,c;
int dfs1(int u)
{for (int v=1;v<=n;v++)if (c.a[v][u]&&!vis[v]){vis[v]=1;if (!f[v]||dfs1(f[v])){f[v]=u;return 1;}}return 0;
}
int dfs2(int u,int from)
{for (int v=1;v<=n;v++)if (c.a[v][u]&&!vis[v]){vis[v]=1;if (f[v]==from||(f[v]>from&&dfs2(f[v],from))){f[v]=u;return 1;}}return 0;
}
int main()
{int cnt=0;scanf("%d",&n);a.rd();b.rd();c=b*a.inv();if (!ok){printf("NIE\n");return 0;}for (int i=1;i<=n;i++){for (int j=1;j<=n;j++) vis[j]=0;if (dfs1(i)) cnt++;}if (cnt<n) printf("NIE\n");else{printf("TAK\n");for (int i=1;i<=n;i++){for (int j=1;j<=n;j++) vis[j]=0;dfs2(i,i);}for (int i=1;i<=n;i++) ans[f[i]]=i;for (int i=1;i<=n;i++) printf("%d\n",ans[i]);}
}