Codechef：Two k-Convex Polygons/TKCONVEX

传送门

题解：
$k$个边可以围成凸多边形的条件是最大边小于其他边之和。

那么最后的答案肯定是连续的两段或者一段，前者枚举，后者暴搜。

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;

const int RLEN=1<<18|1;
inline char nc() {
	static char ibuf[RLEN],*ib,*ob;
	(ib==ob) && (ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
	return (ib==ob) ? -1 : *ib++;
}
inline int rd() {
	char ch=nc(); int i=0,f=1;
	while(!isdigit(ch)) {if(ch=='-')f=-1; ch=nc();}
	while(isdigit(ch)) {i=(i<<1)+(i<<3)+ch-'0'; ch=nc();}
	return i*f;
}

const int N=1e3+50;
struct atom {int a,b;} a[N];
int n,k,ok[N],vis[N]; LL s[N];

inline void dfs(int l,int r,LL s1,LL s2,int c1,int c2,int mx1,int mx2) {
	if(l==r+1) {
		if(s1-mx1>mx1 && s2-mx2>mx2) {
			puts("Yes");
			for(int i=r-2*k+1;i<=r;i++) if(vis[i]==1) printf("%d ",a[i].b);
			for(int i=r-2*k+1;i<=r;i++) if(vis[i]==2) printf("%d ",a[i].b);
			exit(0);
		}
	}
	if(c1!=k) vis[l]=1,dfs(l+1,r,s1+a[l].a,s2,c1+1,c2,a[l].a,mx2);
	if(c2!=k) vis[l]=2,dfs(l+1,r,s1,s2+a[l].a,c1,c2+1,mx1,a[l].a);
}
inline bool solve(int l,int r) {dfs(l,r,0,0,0,0,0,0);} 
int main() {
	n=rd(), k=rd();
	for(int i=1;i<=n;i++) a[i].a=rd(), a[i].b=i;
	sort(a+1,a+n+1,[](atom c,atom d) {return c.a<d.a;});
	for(int i=1;i<=n;i++) s[i]=s[i-1]+a[i].a, ok[i]=(i>=k && s[i-1]-s[i-k]>a[i].a);
	for(int i=k;i<=n;i++) for(int j=i+k;j<=n;j++) if(ok[i] && ok[j]) {
		puts("Yes");
		for(int z=0;z<k;z++) printf("%d ",a[i-z].b);
		for(int z=0;z<k;z++) printf("%d ",a[j-z].b);
		return 0;
	}
	for(int i=2*k;i<=n;i++) solve(i-2*k+1,i);
	puts("No");
}