第十二届北航程序设计竞赛决赛网络同步赛: 浪哥的烦恼

题目链接:https://biancheng.love/problem/587/index

题目描述

最近学校大检查，要求上课认真听讲，这让浪哥很是烦恼，因为这样他就不能上课刷题了。但经过不懈的观察，他发现领导检查是有一定的规律的。不妨把每层楼的教室看成一排，从 1 到 n 标号，领导每检查完一个教室，就会从相邻的一个或两个教室中选择一个继续检查，一个教室可能被检查不止一次。

已知时刻 0 领导在 1 号教室，从 i 号教室走到 i+1 号教室需要 ti 时间，从 i+1 号教室走到 i 号教室也需要 ti 时间 (i=1,2,⋯,n−1) ，检查一个教室的时间极短可忽略不计，放学时刻为 m 。

浪哥想要知道，放学之前（包括 m ）的哪些时刻，领导是不可能检查到他所在的 n 号教室的。

输入

第一行包含一个正整数 T ，表示有 T 组测试数据。

接下来依次给出每组测试数据。对于每组测试数据：

第一行包含两个正整数 n 和 m 。

第二行包含 n−1 个正整数 t1,t2,⋯,tn−1 。

1≤T≤200,2≤n≤100,1≤m≤500,1≤ti≤m

输出

对于每组数据输出一行，包含一些正整数，按升序给出，表示 n 号教室不可能被检查的时刻。

对于本题，输出中在一行的每个整数之间用恰好一个空格隔开，不能有其他额外空格。

输入样例

1
3 10
2 3

输出样例

1 2 3 4 6 7 8 10

样例解释

在放学前，领导有两种不同的检查方式会检查到 3 号教室，分别是 1 -> 2 -> 3 和 1 -> 2 -> 1 -> 2 -> 3 ，其中到达 3 号教室的时刻分别为 5 和 9 。

这道题有两种做法,第一种就是直接用母函数,三个for循环,这个for也不是全部循环,而且范围较小,所以不会超时,代码如下:

//    >File Name: bianchenglove1.cpp
//    > Author: Webwei

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int n,m,vis[505];
int a[105];
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	int t;
	cin>>t;
	while(t--)
	{
		cin>>n>>m;
		memset(vis,0,sizeof(vis));
		int s=0;
		for(int i=1;i<n;i++)
		{
			cin>>a[i];
			s+=a[i];
			a[i]*=2;
			if(s>m) break;
		}
		if(s<=m){
			vis[s]=1;
			for(int i=1;i<n;i++){
				for(int j=s;j<m;j++){
					for(int k=a[i];k+j<=m;k+=a[i])
					{
						if(!vis[k+j]) vis[k+j]=vis[j];
					}
				}
			}
		}
		int flag=0;
		for(int i=1;i<=m;i++)
		{
			if(!vis[i]){
				if(flag) cout<<" ";
				else flag=1;
				cout<<i;
			}
		}
		cout<<'\n';
	}
	return 0;
}

还有另一种就是dp,记忆化搜索

#include<bits/stdc++.h>
using namespace std;
int tim[505];
int h=0;
int ans[505];
int n,m;
int us[105][505];
int dfs(int k,int t)
{

	if(k==n&&t<=m)  ans[t]=1;
	if(t>m||us[k][t])  return 0;
	if(k!=n)  dfs(k+1,t+tim[k]);
	if(k!=1)  dfs(k-1,t+tim[k-1]);
	us[k][t]=1;
}
int main()
{

	int t;
	cin >> t;
	while( t-- )
	{

		cin >> n >> m ;
		fill(ans , ans + m + 1 , 0);
		for(int i=1;i<n;i++)
		{
			cin>>tim[i];
		}
		memset(us,0,sizeof(us));
		dfs(1,0);
		int flag=0;
		for(int i=1;i<=m;i++){

			if(ans[i]==0){

				if(flag==0){

					cout<<i;
					flag=1;
				}else{

					cout<<" "<<i;
				}
			}
		}
		cout<<'\n';
	}
}