题意:
你可以召唤n只怪兽,当第i只怪兽被召唤出来时,它的基础力量是ai,并且你曾经召唤的怪兽都可以获得bi力量,如果你扔掉这只怪兽,bi的加成不会消失,现在你只能留k只怪兽在场上,问你所有在场怪兽的力量总和最大是多少。
题解:
它的数据范围比较小啊,但是又不是特别小,首先我就排除了搜索和状压,至于dp是有可能的吧,但是我懒得去想。
首先可以确定的是,最终留在场上的k只怪兽的召唤顺序一定是按b从小到大的,并且最后一只怪兽放在所有怪兽走完流程之后才召唤,剩下n-k只怪兽重复地进出第k个位置,这样的获利是最大的。
那么如何确定是哪k只怪兽?我们可以发现,虽然它有力量加成这么一个约束,但是不是对某只怪兽的特定约束,而是对位置的约束,所以我就没有考虑网络流。并且只要一只怪兽确定位置的话,它前面和后面的怪兽是谁都无所谓,因此,每只怪兽可以独立地考虑。
所以不妨假设一开始k只怪兽是前k只怪兽,然后从k+1只怪兽到第n只怪兽一个一个考虑。对于每一只新来的怪兽,找到里面可以替换掉的最优解。
我就直接暴力删掉每一只怪兽,然后再for一遍s1(答案集合)看看和是多少。
sum维护的是不在答案集合中的怪兽的b值。
时间复杂度尚可
#include<bits/stdc++.h>
using namespace std;
const int N=80;
struct node{
int id,a,b;
bool operator< (const node& x)const {
if(b==x.b){
if(a==x.a)
return id<x.id;
return a<x.a;
}
return b<x.b;
}
};
multiset<node>s1,s2,tmp;
int in[N];
int main(){
int t;
scanf("%d",&t);
while(t--){
memset(in,0,sizeof(in));
int n,k,a,b;
scanf("%d%d",&n,&k);
s1.clear(),s2.clear();
for(int i=1;i<=n;i++){
scanf("%d%d",&a,&b);
if(i<=k)s1.insert({i,a,b});
else s2.insert({i,a,b});
//printf("%d\n",s2.size());
}
//printf("%d\n",(*s1.begin()).id);
int ans=0,sum=0;
multiset<node>::iterator it,iit;
int i;
for(i=1,it=s1.begin();i<=k;i++,it++)
ans+=(i-1)*(*it).b+(*it).a;
for(it=s2.begin();it!=s2.end();it++)
ans+=(k-1)*(*it).b,sum+=(*it).b;
for(int tim=1;tim<=n-k;tim++){
int mx=ans;
node p;
node now=*s2.begin();
s2.erase(s2.begin());
sum-=now.b;
tmp.clear(),tmp=s1;
s1.insert(now);
for(it=tmp.begin();it!=tmp.end();it++){
s1.erase(s1.find(*it));
int val=0;
for(i=1,iit=s1.begin();i<=k;i++,iit++)
val+=(i-1)*(*iit).b+(*iit).a;
val+=(k-1)*(sum+(*it).b);
if(val>mx)
mx=val,p=*it;
s1.insert(*it);
}
if(mx!=ans)
s1.erase(s1.find(p)),ans=mx,sum+=p.b;
else
s1.erase(s1.find(now)),sum+=now.b;
}
printf("%d\n",k+(n-k)*2);
for(i=1,it=s1.begin();i<k;i++,it++)
in[(*it).id]=1,printf("%d ",(*it).id);
for(i=1;i<=n;i++)
if(!in[i]&&i!=(*it).id)
printf("%d -%d ",i,i);
printf("%d\n",(*it).id);
}
return 0;
}