[loj3342]制作菜品

当$n-1\le m$,不妨令$d_{1}\le d_{2}\le...\le d_{n}$,则$(n-1)k\le mk=\sum_{i=1}^{n}d_{i}\le d_{1}+(n-1)d_{n}$
将这个拆成两部分,即$(n-2)k+k$和$(n-2)d_{n}+(d_{1}+d_{n})$,由于后者大于等于前者,所以两项中必然有一项大于等于前者,容易发现一定存在$k\le d_{1}+d_{n}$
那么就可以不断贪心将$d_{1}$和$d_{n}$匹配,每一次必然会至少消除一个数,而最后一次因为总和恰好为$mk$所以必然全部消除
当$m=n-2$,此时合法当且仅当存在一个集合$S\subset \{1,2,...,n\}$使得$\sum_{i\in S}d_{i}=(|S|-1)k$
证明:充分性,如果存在,将两部分分开处理,即有解且已构造得出
必要性,可以证明若存在合法解,必然存在一组使得每次操作使至少一个点变为$0$(显然消除掉一定更优)
构造:将每一个点向消除掉自己的点连无向边,显然这张图不存在大于2的环(考虑环上操作的先后顺序即可)
因此即去除掉重边后,这张图是一棵森林且有至少2棵树(否则有$n-1$次操作),其中任意一棵子树即可作为集合$S$
考虑怎么求出这个集合$S$,暴力$dp$令$f[i][j][k]$表示前$i$个点中选$j$个点和能否为$k$,复杂度为$o(n^{3}k)$无法通过
如何使得其与$|S|$无关,即$S$需满足$\sum_{i\in S}d_{i}-k=-k$,同时权值范围仅变为$[-nk,nk]$,因此复杂度降为$o(n^{2}k)$
问题即一个01背包的存在性判断,可以用$bitset$来优化,复杂度降为$o(\frac{n^{2}k}{32})$,可以通过
 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 #define N 505
 4 set<pair<int,int> >s;
 5 int t,n,m,k,d[N],vis[N];
 6 bitset<N*N*20>f[N];
 7 void work(int m){
 8     for(int i=1;i<=m;i++){
 9         int x=(*s.begin()).second;
10         s.erase(s.begin());
11         if (d[x]>=k){
12             printf("%d %d",x,k);
13             d[x]-=k;
14             if (d[x])s.insert(make_pair(d[x],x));
15         }
16         else{
17             int y=(*(--s.end())).second;
18             s.erase(--s.end());
19             printf("%d %d %d %d",x,d[x],y,k-d[x]);
20             d[y]-=k-d[x];
21             if (d[y])s.insert(make_pair(d[y],y));
22             d[x]=0;
23         }
24         if (i!=m)printf("\n");
25     }
26 }
27 int main(){
28     freopen("dish.in","r",stdin);
29     freopen("dish.out","w",stdout);
30     scanf("%d",&t);
31     bool flag=0;
32     while (t--){
33         if (flag)printf("\n");
34         flag=1;
35         scanf("%d%d%d",&n,&m,&k);
36         for(int i=1;i<=n;i++)scanf("%d",&d[i]);
37         if (n-1<=m){
38             for(int i=1;i<=n;i++)s.insert(make_pair(d[i],i));
39             work(m);
40             continue;
41         }
42         memset(vis,0,sizeof(vis));
43         for(int i=0;i<=n;i++)f[i].reset();
44         f[0][n*k]=1;
45         bool flagg=0;
46         for(int i=1;i<=n;i++){
47             f[i]=f[i-1];
48             if (d[i]>=k)f[i]|=(f[i-1]<<d[i]-k);
49             else f[i]|=(f[i-1]>>k-d[i]);
50             if (f[i][(n-1)*k]){
51                 flagg=1;
52                 for(int j=i,t=(n-1)*k;j;j--)
53                     if (f[j-1][t-(d[j]-k)]){
54                         vis[j]=1;
55                         t-=d[j]-k;
56                     }
57                 for(int j=1;j<=n;j++)
58                     if (vis[j])s.insert(make_pair(d[j],j));
59                 work(s.size()-1);
60                 printf("\n");
61                 for(int j=1;j<=n;j++)
62                     if (!vis[j])s.insert(make_pair(d[j],j));
63                 work(s.size()-1);
64                 break;
65             }
66         }
67         if (!flagg)printf("-1");
68     }
69     return 0;
70 }
View Code

 

posted @ 2020-08-20 22:44  PYWBKTDA  阅读(109)  评论(0编辑  收藏  举报