题意:给你k,然后给你3*k个数编号为(1~3*k),将它们分成三分,每份k个数,要求至少有两份之和大于500*k,输出他们的编号,前两份的和大于500*k。
思路:玄学题+1!我们肯定挑选最大的2*k个数字,然后对他们进行合理的分配使得两份都大于500*k。随机第一份中的一个位置,随机第二份中的一个位置,然后交换他们,直到两份之和都满足要求。
直接DFS+剪枝超时。
//#include <bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
#include<ctime>
using namespace std;
const int MAXN = 65*3;
struct Node
{
int v, id;
bool operator < (const Node& rhs) const
{
return v > rhs.v;
}
}a[MAXN];
int k;
int main()
{
while (~scanf("%d", &k))
{
for (int i = 0; i < 3*k; i++)
{
scanf("%d", &a[i].v);
a[i].id = i+1;
}
sort(a, a+3*k);
int sum1 = 0, sum2 = 0;
for (int i = 0; i < k; i++)
{
sum1 += a[i].v;
sum2 += a[i+k].v;
}
srand(unsigned(time(NULL)));
while (sum1 <= 500*k || sum2 <= 500*k)
{
int AC1 = rand()%k, AC2 = rand()%k+k;
sum1 = sum1 - a[AC1].v + a[AC2].v;
sum2 = sum2 + a[AC1].v - a[AC2].v;
swap(a[AC1], a[AC2]);
}
for (int i = 0; i < 3*k; i++) printf("%d\n", a[i].id);
}
return 0;
}
/*
2
510
500
500
670
400
310
*/