题目背景
明天就是校园活动了,小明作为场地的负责人,将一切都布置好了。但是在活动的前几天,校园里的灯却都坏掉了,无奈之下,只好再去买一批灯。但是很遗憾的是,厂家看马上要过年了,就没有在进货了,现在只剩下n个发光值不同的灯,作为负责人,你需要,想办法配出合适的灯。
题目描述
厂家有n盏剩下的灯,小明需要m盏灯,因为活动举办在晚上,所以这些灯的光值和不能低于k,现在小明想知道,有多少种选灯的方案,以及每种方案选出的m盏灯。
输入输出格式
输入格式:
共n+1行,第一行有三个整数:n,m,k,表示厂家有n盏(灯),小明需要m盏,从n盏中选的m盏的光线和不能小于k。
接下来的1行,共n个整数,第i个数表示第i盏灯的发光值。
输出格式:
先输出一个整数,表示方案数,接下来的几行,每行m+1个数,前m个数表示每个方案选择的灯的序号,第m+1个数表示这个方案的光值和。
如果没有方案,第一行就输出-1,第二行输出最大的光亮值。
输入输出样例
输入样例#1:
3 2 5
1 3 2
输出样例#1:
1
2 3 5
输入样例#2:
4 2 7
2 5 3 1
输出样例#2:
2
1 2 7
2 3 8
输入样例#3:
5 2 5
2 2 2 2 2
输出样例#3:
-1
4
说明
- 数据说明
3<=m<=n<=20;
1<=k<=a[1]+a[2]...+a[n];
样例就自己看哈~。
输出用序号排序
思路:搜索。
错因:要加上一个记录前驱,否则会搜索重复。
#include<iostream> #include<cstdio> #include<algorithm> #include<cstring> using namespace std; int n,m,k,ans,w[21],vis[21],v[10000][21]; void dfs(int now,int num,int pre){ if(now==m){ if(num>=k){ ans++; memcpy(v[ans],vis,sizeof(vis)); v[ans][0]=num; } return ; } for(int i=pre;i<=n;i++) if(!vis[i]){ vis[i]=1; dfs(now+1,num+w[i],i+1); vis[i]=0; } } int main(){ scanf("%d%d%d",&n,&m,&k); for(int i=1;i<=n;i++) scanf("%d",&w[i]); dfs(0,0,1); if(ans==0){ sort(w+1,w+1+n); for(int i=n;i>=n-m+1;i--) ans+=w[i]; cout<<"-1"<<endl<<ans; return 0; } cout<<ans<<endl; for(int i=1;i<=ans;i++){ for(int j=1;j<=n;j++) if(v[i][j]) cout<<j<<" "; cout<<v[i][0]<<endl; } }