原题链接
假设当前排列为(p1,p2,p3…,pn)
那么最终需要位置数为:
m
a
x
(
p
1
,
p
2
)
+
m
a
x
(
p
2
,
p
3
)
+
.
.
.
+
m
a
x
(
p
n
−
1
,
p
n
)
+
m
a
x
(
p
n
,
p
1
)
max(p_1,p_2)+max(p_2,p_3)+...+max(p_{n-1},p_n)+max(p_n,p_1)
max(p1,p2)+max(p2,p3)+...+max(pn−1,pn)+max(pn,p1)
我们可以采取一个贪心的策略,将p按照降序排列,这样可以得到上诉式子的下界。
证明:交换pi,pi+1,其它位置不变
原式子为:
S
1
=
m
a
x
(
p
i
−
1
,
p
i
)
+
m
a
x
(
p
i
,
p
i
+
1
)
+
m
a
x
(
p
i
+
1
,
p
i
+
2
)
=
p
i
+
p
i
+
1
+
p
i
+
2
S1=max(p_{i-1},pi)+max(pi,p_{i+1})+max(p_{i+1},p_{i+2})=p_i+p_{i+1}+p_{i+2}
S1=max(pi−1,pi)+max(pi,pi+1)+max(pi+1,pi+2)=pi+pi+1+pi+2
变化式子为:
S
2
=
m
a
x
(
p
i
−
1
,
p
i
+
1
)
+
m
a
x
(
p
i
+
1
,
p
i
)
+
m
a
x
(
p
i
,
p
i
+
2
)
=
2
∗
p
i
+
1
+
p
i
+
2
S2=max(p_{i-1},p_{i+1})+max(p_{i+1},p_i)+max(p_i,p_{i+2})=2*p_{i+1}+p_{i+2}
S2=max(pi−1,pi+1)+max(pi+1,pi)+max(pi,pi+2)=2∗pi+1+pi+2
由于
p
i
<
=
p
i
+
1
p_{i}<=p_{i+1}
pi<=pi+1所以S1<=S2,结论得证.
每个人还需要1个位置,可知最终所需位置为 n+∑ni=1(ai)−min(ai)+max(ai)≤m. (ai={a1,a2,…an})
#include <bits/stdc++.h>
using namespace std;
int main()
{
ios_base::sync_with_stdio(false);
int t;
cin >> t;
while (t--) {
long long n, m;
cin >> n >> m;
long long sum=0, min_val=1e9, max_val=0;
for (int i=1; i<=n; i++) {
long long x;
cin >> x;
sum+=x, min_val=min(min_val, x), max_val=max(max_val, x);
}
cout << (n+sum-min_val+max_val<=m ? "YES" : "NO") << "\n";
}
return 0;
}