题意:给一个序列,通过交换元素位置,得到一个尽量小的序列,注意只有g[i][j]=='1'时,a[i]和a[j]才能交换位置。
主要是判断出哪些位置的元素可以交换,用Floyd和并查集都行,当然,后者要快些。
Floyd的代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#include<queue>
#include<vector>
#include<set>
using namespace std;
int a[100010],n;
char s[410][410];
int main()
{
scanf("%d",&n);
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
for(int i=0;i<n;i++)
scanf("%s",s[i]);
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
for(int k=0;k<n;k++)
if(s[j][i]=='1'&&s[i][k]=='1')
s[j][k]='1';
for(int i=0;i<n;i++)
for(int j=i+1;j<n;j++)
if(s[i][j]=='1'&&a[i]>a[j])
swap(a[i],a[j]);
for(int i=0;i<n-1;i++)
printf("%d ",a[i]);
printf("%d\n",a[n-1]);
return 0;
}
下面是另一个Floyd的代码,数据的交换虽然复杂了但挺巧妙的。
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<queue>
#include<stack>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
const int M=1010;
const int N=310;
const int INF=1e9;
const double PI=acos(-1.0);
const double Exp=exp(1.0);
int n,m;
int pos[N],val[N],g[N][N];
void init()
{
int i,j;
int t;
for (i=1;i<=n;i++) {
scanf("%d%*c",&t);
val[i]=t;
pos[t]=i;
} //getchar();
for (i=1;i<=n;i++){
for (j=1;j<=n;j++)
{
char c;
scanf("%c",&c);
g[i][j]=c-'0';
}
getchar();
}
}
void floyd()
{
for (int k=1;k<=n;k++)
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++) if (g[i][j]==0) g[i][j]=g[i][k] & g[k][j];
}
int main()
{
int i,j;
while(~scanf("%d",&n))
{
init();
floyd();
for (i=1;i<=n;i++) // pos
for (j=1;j<=n;j++) if (g[i][pos[j]]) { // val
int _pos=pos[j]; //cout<<i<<" "<<pos[j]<<": "; for (int k=1;k<=n;k++) cout<<val[k]<<" ";cout<<endl; //cout<<i<<" "<<pos[j]<<endl; if (j==n) cout<<"end"<<endl;
val[_pos]=val[i];
pos[val[i]]=_pos;
val[i]=j;
pos[j]=i;
//break;
}
for (i=1;i<=n;i++) {
if (i<=n-1) printf("%d ",val[i]);
else printf("%d\n",val[i]);
}
}
}
并查集的代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<queue>
#include<stack>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
const int M=1010;
const int N=310;
const int INF=1e9;
const double PI=acos(-1.0);
const double Exp=exp(1.0);
int n,m;
int pos[N],val[N],g[N][N],f[N];
int Find(int x)
{
return x==f[x]?x:(f[x]=Find(f[x]));
}
void init()
{
int i,j;
int t;
for (i=1;i<=n;i++) {
scanf("%d%*c",val+i);
f[i]=i;
}
for (i=1;i<=n;i++){
for (j=1;j<=n;j++)
{
char c;
scanf("%c",&c);
g[i][j]=c-'0';
if (g[i][j]) {
int a=Find(i),b=Find(j);
f[b]=a;
}
}
getchar();
}
}
int main()
{
int i,j;
while(~scanf("%d",&n))
{
init();
for (i=1;i<=n;i++){
for (j=i+1;j<=n;j++){
int a=Find(i),b=Find(j); //printf("(%d %d) (%d %d)\n",i,a,j,b);
if (a==b && val[i]>val[j]){
swap(val[i],val[j]);
}
}
}
for (i=1;i<=n;i++) {
if (i<=n-1) printf("%d ",val[i]);
else printf("%d\n",val[i]);
}
}
}