贪心、dfs--Codechef TKCONVEX

本文主要是介绍贪心、dfs--Codechef TKCONVEX，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

题目大意：
n 根木棍长度分别为ai，现在想从中选出2k 根组成两个面积大于0 的
凸k 边形，请找出一组解或判断无解
2k<=n<=1000 , 3<=k<=10 , 1<=ai<=109

solution：
k 条边成为一个凸多边形的充要条件是最长边小于其他边之和
将木棍按长度排序后，按上述条件可知可行解要么是两段不相交的
连续的子序列，要么是一个连续的长为2k 的子序列
使用前缀和可以O(n) 判断出第一种可能
第二种可能可以考虑暴力枚举求解，即枚举子序列的左端点，右端点随之确定，剩下的问题是分配木棍，由于k 最多只有10，所以暴搜即可

嗯其实很简单但我不会告诉你我暴搜都能写挂
好吧其实是没怎么看题，题上让输出的是数字在原序列中的标号
结果我直接一排序就忘记记录原标号了你懂得···
wa了十几次不想说话mdzz

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#define maxn 1010
#define LL long long
using namespace std;
int n,k,cnt,last,l[5],r[5],mxp;
LL sum[maxn];
vector<int> ans[5];
bool flg;struct qwq{int id; LL w;
}a[maxn];
bool cmp(qwq x,qwq y) {return x.w<y.w;}
inline LL rd(){LL x=0,f=1;char c=' ';while(c<'0' || c>'9') {if(c=='-')f=-1;c=getchar();}while(c<='9' && c>='0') x=x*10+c-'0',c=getchar();return x*f;
}void dfs(int p,LL sum1,LL sum2){if(flg || p>mxp+1) return;if(ans[1].size()==k && ans[2].size()==k) {if(sum1-a[ans[1][k-1]].w<=a[ans[1][k-1]].w || sum2-a[ans[2][k-1]].w<=a[ans[2][k-1]].w) return;printf("Yes\n"); flg=true;for(int i=0;i<ans[1].size();i++) printf("%d ",a[ans[1][i]].id);for(int i=0;i<ans[2].size();i++) printf("%d ",a[ans[2][i]].id);return;}if(ans[2].size()<k){ans[2].push_back(p);dfs(p+1,sum1,sum2+a[p].w);ans[2].pop_back();}if(ans[1].size()<k){ans[1].push_back(p);dfs(p+1,sum1+a[p].w,sum2);ans[1].pop_back();}return;
}int main(){n=rd(); k=rd(); int last=0,cnt=0;for(int i=1;i<=n;i++) a[i].w=rd(),a[i].id=i;sort(a+1,a+n+1,cmp);for(int i=1;i<=n;i++){sum[i]=sum[i-1]+a[i].w;if(i-k+1>last && i>=k && a[i].w<sum[i-1]-sum[i-k]){cnt++; last=i; l[cnt]=i-k+1, r[cnt]=i;}if(cnt==2) break;}if(cnt==2){printf("Yes\n");for(int i=l[1];i<=r[1];i++) printf("%d ",a[i].id);for(int i=l[2];i<=r[2];i++) printf("%d ",a[i].id);return 0;}for(int i=1;i+2*k-1<=n;i++){mxp=i+2*k-1;dfs(i,0,0);if(flg) break;}if(!flg) printf("No\n");return 0;
}