F. Vitaly and Advanced Useless Algorithms---01背包求方案数

最新推荐文章于 2024-10-06 10:17:48 发布

_WAWA鱼_

最新推荐文章于 2024-10-06 10:17:48 发布

阅读量317

点赞数

分类专栏： DP 文章标签：动态规划 c++ 算法

本文链接：https://blog.csdn.net/qq_51282224/article/details/123386409

版权

DP 专栏收录该内容

48 篇文章 3 订阅

订阅专栏

该代码实现了一个动态规划算法来解决一个任务调度问题。输入包括任务数、任务依赖和每个任务的执行时间，输出是满足限制条件的最短执行路径。程序首先读取任务数量和每个任务的限制值，然后构建任务依赖矩阵并应用动态规划策略找到最优解。如果总执行时间超过限制，则返回-1。否则，输出删除的任务数量及其ID。

摘要由CSDN通过智能技术生成

题目链接

在这里插入图片描述

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#include <array>
using namespace std;
#define int long long

void solve()
{
	int n,m;
	cin>>n>>m;
	vector<int>lim(n);
	for(int i=0;i<n;i++)cin>>lim[i];
	vector<vector<array<int,3>>> a(n);
	for(int i=0;i<m;i++)
	{
		int e,t,p;
		cin>>e>>t>>p;
		e--;
		a[e].push_back({t,p,i});
	}
	
	vector<int>res;
	int sum=0;
	for(int i=0;i<n;i++)
	{
		int N = a[i].size();
		vector dp(N+1,vector<int>(101,2e9+1));
		dp[0][0]=0;
		for(int j=0;j<N;j++)
		{
			dp[j+1]=dp[j];
			auto [t,p,_] = a[i][j];
			for(int k=0;k<=100;k++)
			{
				int nk=min(100ll,k+p);
				dp[j+1][nk]=min(dp[j+1][nk],dp[j][k]+t);
			}
		}
		
		sum+=dp[N][100];
		if(sum>lim[i])
		{
			cout<<-1<<'\n';
			return ;
		}
		
		int nd=100;
		for(int j=N-1;j>=0;j--)
		{
			auto [t,p,id] = a[i][j];
			if(dp[j+1][nd]==dp[j][nd])continue;
			res.push_back(id);
			for(int k=0;k<=100;k++)
			{
				int nk=min(100ll,k+p);
				if(nk==nd&&dp[j][k]+t==dp[j+1][nk])
				{
					nd=k;
					break;
				}
			}
		}
	}
	cout<<res.size()<<'\n';
	for(auto x : res)cout<<x+1<<" ";
	cout<<'\n';
}

signed main()
{
    int T;cin>>T;
	while(T--)solve(); 
}