C题
题目大意
题目链接
在公司中员工相互管理,求某个员工最年轻的直接或者间接管理者。
分析
员工之间的相互管理可以看成一个有向图,如果用一个邻接矩阵来存储的话会造成空间的浪费并且时间复杂度也会变高,所以这里用邻接表来存储顶点之间的关系。在改变员工A,B的位置的时候先交换A,B员工所在矩阵的行(交换两个人的管理者),再遍历整个矩阵,如果有员工的管理者是A或者B就相互改变,简而言之就是交换A,B两个人的管理者和手下。再来说说查询年龄,这里用到dfs来搜索最小年龄,除了第一个员工,其他查询员工的年龄都要和自己的管理者进行比较(选择最小的),所以我在第一次搜索的时候就初始化result为无穷大,其他时候就把result初始化为这次搜索的员工的年龄。其次在多次搜索的时候会有重复的,增大了时间复杂度,所以我用一个二维数组来存储搜索结果,如果已经搜索过了,就直接返回二维数组中存储的值,不必再进行重复的搜索(这里要注意一旦交换两个员工的位置,搜索结果就会发生改变,所以在交换两个员工位置之后就要将记忆数组给清空)。
代码
#include <stdio.h>
#include <iostream>
#include <vector>
#include <string.h>
#include <stdlib.h>
#include <algorithm>
#define INF 0x3f3f3f3f
#define MAX_N 500+10
using namespace std;
vector<int>G[MAX_N];
int number[MAX_N];
int dp[MAX_N][MAX_N];
int dfs(int num,int cnt)
{
if(dp[num][cnt]>=1)
{
return dp[num][cnt];
}
int result;
if(cnt==0)
result=INF;
else
result=number[num];
if(G[num].size()==0)
{
result=number[num];
dp[num][cnt]=result;
return result;
}
else
{
for(int i=0;i<G[num].size();i++)
{
result=min(dfs(G[num][i],1),result);
}
}
return dp[num][cnt]=result;
}
int main(int argc, char const *argv[])
{
int N,M,I;
scanf("%d%d%d",&N,&M,&I);
for(int i=1;i<=N;i++)
scanf("%d",&number[i]);
int a,b;
for(int i=0;i<M;i++)
{
scanf("%d%d",&a,&b);
G[b].push_back(a);
}
while(I--)
{
char d;
getchar();
scanf("%c",&d);
if(d=='P')
{
int num;
scanf("%d",&num);
if(G[num].size()==0)
{
printf("*\n");
continue;
}
else
{
int result1=dfs(num,0);
printf("%d\n",result1);
continue;
}
}
else if(d=='T')
{
int temp;
memset(dp,0,sizeof(dp));//一旦交换员工的位置,dp的结果就会发生改变
vector<int>mid;
mid.clear();//每一次交换矩阵两行的时候长度都不一样,所以要清空。
int num1,num2;
scanf("%d%d",&num1,&num2);
for(int i=0;i<G[num1].size();i++)
{
mid.push_back(G[num1][i]);
}
G[num1].clear();
for(int i=0;i<G[num2].size();i++)
{
G[num1].push_back(G[num2][i]);
}
G[num2].clear();
for(int i=0;i<mid.size();i++)
{
G[num2].push_back(mid[i]);
}
for(int i=1;i<=N;i++)
{
for(int j=0;j<G[i].size();j++)
{
if(G[i][j]==num1)
G[i][j]=num2;
else if(G[i][j]==num2)
G[i][j]=num1;
}
}
}
}
return 0;
}
总结
最开始的时候交换矩阵两行的时候想偷懒,想用指针直接交换两行的值,结果运行程序会崩,所以最终还是先用一个数组存储下矩阵的一行,遍历来交换矩阵的两行(切记在竞赛中最好不要用指针)。其次在这道题中,邻接表,无穷大INF,dfs都得到了很大的用处。