贪心、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;
}