傳送門:https://cn.vjudge.net/problem/Gym-100531B
注意 :
對k個狀態進行枚舉就可以了
第一組測試樣例中的輸出順序是可以換的,沒有要求
需要用到 long long
和需要用數組存不然避免超時
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
struct node
{
int x;
int id;
} a[120000], b[120000];//數組需要開的大一點兒
long long int sum[120000];//第一種直接加的求和
long long int ans[120000];//第二種百分比的求和
int cmp ( node a,node b)
{
return a.x>b.x;
}
double solve(int n, long long int x,long long int y)// 注意,x y的數據類型 long long
{
double k=((n+x)*(100+y)/100.0);
return k;
}
int main ()
{
freopen("buffcraft.in", "r", stdin);
freopen("buffcraft.out", "w", stdout);
int n,m,x,y;
scanf("%d %d %d %d",&n,&m,&x,&y);
for(int i=0; i<x; i++)
{
scanf("%d",&a[i].x);
a[i].id=i+1;
}
for(int i=0; i<y; i++)
{
scanf("%d",&b[i].x);
b[i].id=i+1;
}
sort(a,a+x,cmp);
sort(b,b+y,cmp);
sum[0]=a[0].x;
ans[0]=b[0].x;
for(int i=1;i<x;i++)
{
sum[i]=sum[i-1]+a[i].x; //存到沒一狀態的和
}
for(int i=1;i<y;i++)
{
ans[i]=ans[i-1]+b[i].x;
}
int p=0,q=0;
double M;
M=0;
for(int i=0; i<=x; i++)
{
if (i>m) break; //注意 如果break ,會多跑很多超時
int j=m-i;
if(j>y) j=y; // j>y 並不能取break,因爲很可能此時的第一種還可以多取,第二種只不過只能取y種
long long sum1, sum2;
if(i==0) sum1=0;
else sum1=sum[i-1];
if(j==0) sum2=0;
else sum2=ans[j-1];
double Max=solve(n,sum1,sum2);
if(Max>M)
{
M=Max;
p=i;
q=j;
}
}
printf("%d %d\n",p,q);
for(int i=0; i<p; i++)
{
printf("%d%c",a[i].id, i==p-1?'\n':' ');
}
for(int i=0; i<q; i++)
{
printf("%d%c",b[i].id, i==q-1?'\n':' ');
}
return 0;
}