Description
A城市有一个巨大的圆形广场,为了绿化环境和净化空气,市政府决定沿圆形广场外圈种一圈树。园林部门得到指令后,初步规划出n个种树的位置,顺时针编号1到n。并且每个位置都有一个美观度Ai,如果在这里种树就可以得到这Ai的美观度。但由于A城市土壤肥力欠佳,两棵树决不能种在相邻的位置(i号位置和i+1号位置叫相邻位置。值得注意的是1号和n号也算相邻位置!)。最终市政府给园林部门提供了m棵树苗并要求全部种上,请你帮忙设计种树方案使得美观度总和最大。如果无法将m棵树苗全部种上,给出无解信息。
Input
输入的第一行包含两个正整数n、m。第二行n个整数Ai。
Output
输出一个整数,表示最佳植树方案可以得到的美观度。如果无解输出“Error!”,不包含引号。
Sample
Input
【样例输入1】
7 3
1 2 3 4 5 6 7
【样例输入2】
7 4
1 2 3 4 5 6
7
Sample
Output
【样例输出1】
15
【样例输出2】
Error!
【数据规模】
对于全部数据:m<=n;
-1000<=Ai<=1000
N的大小对于不同数据有所不同:
数据编号
N的大小 数据编号 N的大小
1 30 11 200
2 35 12 2007
3 40 13 2008
4 45 14
2009
5 50 15 2010
6 55 16 2011
7 60 17 2012
8 65 18 199999
9 200
19 199999
10 200 20 200000
看wikioi的题解完全没有看懂
于是问了群里的人,结果告诉我是费用流优化
搞了半天总算懂了
首先我们把它拆成链,因为这样我们就好做了,我们只要算两遍就行了,一个是[1,n-1](表示n不取)一个是[2,n](表示1不取)
然后构费用流的图(有点奇葩......)
然后跑费用流就行了
但是显然不能跑费用流,承受不了
所以我们模拟费用流
假设我们选了点2(因为费用最大),然后增广后,残留网络是这样的
第1,2,3条路都没了
但是我们可以走从1到2到3,即多了一条路,费用为a[1]+a[3]-a[2]
我们也可以分析出只多出了这条路,其他的路都是没有用的
因为a[2]>=a[1],所以我们不可能从1走到2再走到S,所以没有别的路了
所以对于这种情况我们就用a[pre[i]]+a[next[i]]-a[i]代替a[i],然后删除pre[i]和next[i]
还有一种情况就是i是两端
这种情况想一想就知道了,端点是最大的那么就选端点,然后删掉在端点的那两个值就行了
所以我们用堆维护最大值,做m次就行了
1 const
2 maxn=200010;
3 var
4 q,h,a,b,pre,next:array[0..maxn]of longint;
5 n,m,tot:longint;
6
7 function max(x,y:longint):longint;
8 begin
9 if x>y then exit(x);
10 exit(y);
11 end;
12
13 procedure swap(var x,y:longint);
14 var
15 t:longint;
16 begin
17 t:=x;x:=y;y:=t;
18 end;
19
20 procedure up(x:longint);
21 var
22 i:longint;
23 begin
24 while x>1 do
25 begin
26 i:=x>>1;
27 if b[q[x]]>b[q[i]] then
28 begin
29 swap(q[x],q[i]);
30 h[q[x]]:=x;
31 h[q[i]]:=i;
32 x:=i;
33 end
34 else break;
35 end;
36 end;
37
38 procedure down(x:longint);
39 var
40 i:longint;
41 begin
42 i:=x<<1;
43 while i<=tot do
44 begin
45 if (i<tot) and (b[q[i+1]]>b[q[i]]) then inc(i);
46 if b[q[i]]>b[q[x]] then
47 begin
48 swap(q[x],q[i]);
49 h[q[x]]:=x;
50 h[q[i]]:=i;
51 x:=i;
52 i:=x<<1;
53 end
54 else break;
55 end;
56 end;
57
58 procedure insert(x:longint);
59 begin
60 inc(tot);
61 h[x]:=tot;
62 q[tot]:=x;
63 up(tot);
64 end;
65
66 procedure delete(x:longint);
67 begin
68 if x=0 then exit;
69 b[q[x]]:=maxlongint;
70 up(x);
71 swap(q[1],q[tot]);
72 h[q[1]]:=1;
73 h[q[tot]]:=tot;
74 dec(tot);
75 down(1);
76 end;
77
78 function f(l,r:longint):longint;
79 var
80 i,u,v,t:longint;
81 begin
82 f:=0;
83 tot:=0;
84 for i:=l to r do
85 begin
86 b[i]:=a[i];
87 insert(i);
88 pre[i]:=i-1;
89 next[i]:=i+1;
90 end;
91 pre[l]:=0;
92 next[r]:=0;
93 for i:=1 to m do
94 begin
95 t:=q[1];
96 inc(f,b[t]);
97 u:=pre[t];
98 v:=next[t];
99 if (u<>0) and (v<>0) then
100 begin
101 b[t]:=b[u]+b[v]-b[t];
102 pre[t]:=pre[u];
103 next[t]:=next[v];
104 next[pre[t]]:=t;
105 pre[next[t]]:=t;
106 down(h[t]);
107 end
108 else
109 begin
110 if u<>0 then next[pre[u]]:=0;
111 if v<>0 then pre[next[v]]:=0;
112 delete(h[t]);
113 end;
114 delete(h[u]);
115 delete(h[v]);
116 end;
117 end;
118
119 procedure main;
120 var
121 i:longint;
122 begin
123 read(n,m);
124 if m*2>n then
125 begin
126 writeln(‘Error!‘);
127 exit;
128 end;
129 for i:=1 to n do
130 read(a[i]);
131 writeln(max(f(1,n-1),f(2,n)));
132 end;
133
134 begin
135 main;
136 end.