zr#954 分组

分析

我们按照w排序

之后2是左括号1是右括号

3既可以是左括号又可以是右括号

然后直接dp即可

我们发现合法的k不会超过sqrt

所以复杂度是对的

对于w相等的情况我们按照种类231排序

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<ctime>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
#define int long long
#define up(x,y) x=min(x,y)
const int inf = 1e16;
struct node {
  int w,s,p;
};
node d[100100];
int n,m,dp[2][2020][1010],now,wh[10];
inline bool cmp(const node a,const node b){
    if(a.w==b.w){
      return a.p<b.p;
    }
    return a.w<b.w;
}
signed main(){
    int i,j,k;
    scanf("%lld%lld",&n,&m);
    wh[1]=3;
    wh[2]=1;
    wh[3]=2;
    for(i=1;i<=n;i++){
      scanf("%lld%lld%lld",&d[i].w,&d[i].s,&d[i].p);
      d[i].p=wh[d[i].p];
    }
    if(2*m>n){
      puts("-1");
      return 0;
    }
    sort(d+1,d+n+1,cmp);
    memset(dp,42,sizeof(dp));
    dp[0][0][0]=0ll;
    int pr=1;now=0;
    for(i=1;i<=n;i++){
      now^=1;pr^=1;
      for(j=0;j<=m+m;j++)
        for(k=0;k<=m&&k<=j;k++){
          up(dp[now][j][k],dp[pr][j][k]);
          if(d[i].p==1||d[i].p==2){
              if(k<m){
                up(dp[now][j+1][k+1],dp[pr][j][k]+d[i].s);
              }
          }
          if(d[i].p==2||d[i].p==3){
              if(k>0){
              up(dp[now][j+1][k-1],dp[pr][j][k]+d[i].s);
              }
          }
          dp[pr][j][k]=inf;
        }
    }
    printf("%lld\n",(dp[now][m+m][0]>=inf?-1:dp[now][m+m][0]));
    return 0;
}