题意
n个人在n个城市之间跑马拉松,每个人的起点不同终点任意但必须走最短路,每个人速度相同,按完成时间和起点编号排名,发g∈[g1,g2]个金牌,发s∈[s1,s2]个金牌,发个n-g-s个铜牌,问有多少种不同的奖牌发放方案。
题解
首先Floyd计算最短路,然后做如下处理:
f[i][j]=f[i][j]*n+i
这样不同选手之间比较不会出现同名次情况。然后维护2个数组:
mind[i]=min(f[i][j])
maxd[i]=max(f[i][j])
然后用mind[i]来枚举金牌线,即i选手以mind[i]的成绩拿了金牌最后一名。这样枚举是不会重复的,因为每次枚举刚好将一名选手从可能拿金牌的池子里筛出去,假设就是i选手被筛出去,那么这次枚举的金牌发放肯定是没有i选手的,而上一次枚举金牌最后一名就是i选手即金牌发放一定有i选手。铜牌线同理使用maxd[j]来枚举。金牌线铜牌线的枚举复杂度是O(n^2)。
每次枚举维护G、S、B三个数组,表示每个选手是否有可能性拿到金银铜牌:
Gk]=1 if mind[k]<mind[i]
S[k]=1 if exist mind[i]<f[k][l]<maxd[j]
B[k]=1 if maxd[k]>maxd[j]
dp[k][x][y]表示这次枚举考虑前k个人分配x个金牌y个银牌的方案数,dp[0][0][0]=1,有转移:
dp[k+1][x+1][y]+=dp[k][x][y] if G[k]==1
dp[k+1][x][y+1]+=dp[k][x][y] if S[k]==1
dp[k+1][x][y]=0 if B[k]==0 # 多了一个人但是金银牌数量不变又不能拿铜牌那就没牌了
注意手动地G[i]=1,S[i]=0,B[i]=0,G[j]=0,S[j]=0,B[j]=1,防止i选手和j选手去拿这次枚举情况下不该拿的牌子。