天池 在线编程 订单分配(回溯)

文章目录

1. 题目

描述
打车派单场景, 假定有N个订单,待分配给N个司机。
每个订单在匹配司机前,会对候选司机进行打分,打分的结果保存在N*N的矩阵score,其中score[i][j]代表订单 i 派给司机 j 的分值。
假定每个订单只能派给一位司机,司机只能分配到一个订单。
求最终的派单结果,使得匹配的订单和司机的分值累加起来最大,并且所有订单得到分配。

在这里插入图片描述

题目保证每组数据的最大分数都是唯一的

示例

样例 1
输入:
[[1,2,4],[7,11,16],[37,29,22]]
输出:
[1,2,0]
解释:
标号为0的订单给标号为1的司机,获得 score[0][1] = 2 分,
标号为1的订单给标号为2的司机,获得 score[1][2] = 16 分,
标号为2的订单给标号为0的司机,获得 score[2][0] = 37 分,
所以一共获得了 2 + 16 + 37 = 55 分。

https://tianchi.aliyun.com/oj/403958541416425656/430542446415516442

2. 解题

class Solution {
public:
    /**
     * @param score: When the j-th driver gets the i-th order, we can get score[i][j] points.
     * @return: return an array that means the array[i]-th driver gets the i-th order.
     */
	int ans = 0;//最大总和
	vector<int> path;//选取的最佳方案
    vector<int> orderAllocation(vector<vector<int>> &score) {
        // write your code here
        int n = score.size();
        vector<bool> vis(n, false);
        vector<int> p(n);
        dfs(score, vis, 0, 0, p);
        return path;
    }
    void dfs(vector<vector<int>> &score, vector<bool>& vis, int i, int val, vector<int>& p)
    {
    	if(i == score.size())
    	{
    		if(val > ans)
    		{
    			ans = val;
    			path = p;
    		}
    		return;
    	}
    	for(int j = 0; j < score.size(); ++j)
    	{
    		if(vis[j]) continue;//物品 j 被选了
    		vis[j] = true;
    		p[i] = j; // 选取的方案 , i 选取了 j 物品
    		dfs(score, vis, i+1, val+score[i][j], p);
    		vis[j] = false;
    	}
    }
};

54ms C++


我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!
Michael阿明

相关推荐
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页