1.题目描述:
A有n个数,B有n个数,要你在这n个位置中选取上限为 大小的子数组,使得它对应a下标子串的和大于 a数组总和的 1/2,且对应b下标的子串的和大于b数组总和的1/2。
3.解题思路:
开始没什么好想法,凌晨脑袋比较乱,只想到dfs去暴力,但是暴力又写x了,结果wa了一晚上,赛后再来了一发dfs,果断t了,确实1e5的数组去暴力dfs很难过。
正解应该往数学方面想,首先题只要你给出满足条件的解,且保证一定有解,其实就是一个鸽巢原理,按最大值排序后选择前 个,假设这 的数都小于sum / 2,很容易证明数组总和是小于sum的,因此反证法证明了一定可以取,下面就直接贪心地取好了。在standing上还看到一种奇妙的方法是利用随机数,这样复杂度不好说,但是确实在大量数据面前很有优势。
4.AC代码:
#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define maxn 100100
#define lson root << 1
#define rson root << 1 | 1
#define lent (t[root].r - t[root].l + 1)
#define lenl (t[lson].r - t[lson].l + 1)
#define lenr (t[rson].r - t[rson].l + 1)
#define N 1111
#define eps 1e-6
#define pi acos(-1.0)
#define e exp(1.0)
using namespace std;
const int mod = 1e9 + 7;
typedef long long ll;
typedef unsigned long long ull;
int a[maxn], b[maxn], p[maxn];
int myrand(int x)
{
return rand() % x;
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("in.txt", "r", stdin);
freopen("out.txt", "w", stdout);
long _begin_time = clock();
#endif
int n;
while (~scanf("%d", &n))
{
srand(time(NULL));
ll suma = 0, sumb = 0;
for (int i = 1; i <= n; i++)
{
scanf("%d", &a[i]);
suma += a[i];
p[i] = i;
}
for (int i = 1; i <= n; i++)
{
scanf("%d", &b[i]);
sumb += b[i];
}
bool flag = 0;
while (!flag)
{
random_shuffle(p + 1, p + n + 1, myrand);
ll sum1 = 0, sum2 = 0;
for (int i = 1; i <= n / 2 + 1; i++)
{
sum1 += a[p[i]];
sum2 += b[p[i]];
if (sum1 > suma / 2 && sum2 > sumb / 2)
{
flag = 1;
break;
}
}
}
printf("%d\n", n / 2 + 1);
for (int i = 1; i <= n / 2 + 1; i++)
if (i == 1)
printf("%d", p[i]);
else
printf(" %d", p[i]);
puts("");
}
#ifndef ONLINE_JUDGE
long _end_time = clock();
printf("time = %ld ms.", _end_time - _begin_time);
#endif
return 0;
}