Codeforces 1354 F Summoning Minions —— 貪心

This way

題意：

你可以召喚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;
}