Codeforces Round 950 (Div. 3)补题(D)
链接:Codeforces Round 950 (Div. 3)
D. GCD-sequence
题目:
解题思路:
先遍历一遍序列a,计算出每个gcd,找到第一个递减的gcd,然后会出现以下两种情况:
- 如果找不到递减的gcd,说明原序列就已经满足条件,那么我们如果减去末尾元素或者第一个元素,构造成的b序列的gcd就相当于a序列的gcd减去最后一个或者第一个,一定满足条件
- 如果找到了递减的gcd,那么记录下他的下标pos,导致递减的元素的位置在a序列中的位置一定在pos-1,pos,pos+1这三个位置中的一个,分别构造三个序列减去这三个位置中的一个,然后在判断是否满足不递减的条件,如果还不满足,则还需删去元素,不能构造出b序列
个人复盘:
自己做的时候第一种情况想到了,但是第2个情况没想清楚,想着直接用cnt来记录导致递减的元素的个数,然后发现要考虑的因素太多了,什么首尾位置的特殊考虑,用该位置 “比后一个大来” 考虑好还是 “比前一个位置小” 来考虑好,如果删除元素是生成的gcd太小导致不满足,还是生成的gcd太大导致后面满足条件的gcd都比他小导致的不满足…结果自己分类讨论把自己cpu干烧了,最后发现只用考虑第一个,然后判断删去后是否存在第二个就行了
有时候如果发现分类讨论太复杂了就该考虑一下是不是自己思考方向错了
代码:
#include <bits/stdc++.h>
using namespace std;
#define ll long long
bool Is_Nonde(vector<ll> &arr) {
ll last_g = __gcd(arr[0], arr[1]);
for (int i = 1; i < arr.size()-1; i++) {
ll g = __gcd(arr[i], arr[i+1]);
if (g < last_g) return false;
last_g = g;
}
return true;
}
void solve() {
int n; cin >> n;
vector<ll> a(n);
for (int i = 0; i < n; i++) cin >> a[i];
vector<ll> b1, b2, b3;
b1 = b2 = b3 = a;
ll last_g = -1, g = -1;
int pos;
//找出第一个递减的位置,记录该位置
for (int i = 0; i < n-1; i++) {
g = __gcd(a[i], a[i+1]);
if (g < last_g) pos = i;
last_g = g;
}
//如果原序列已经满足不递减,则删末尾元素仍不递减
if (pos == -1)
cout << "YES\n";
//如果原序列递减,则导致递减的元素肯定在当前位置及其左右位置中,删除分别删除可能导致递减的元素
if (pos > 0) b1.erase(b1.begin()+pos-1);
b2.erase(b2.begin()+pos);
if (pos < n-1) b3.erase(b3.begin()+pos+1);
//判断删除后还是否递减
if (Is_Nonde(b1) || Is_Nonde(b2) || Is_Nonde(b3)) {
cout << "YES\n";
} else {
cout << "NO\n";
}
}
int main () {
ios_base::sync_with_stdio(0); cin.tie(0);cout.tie(0);
int t; cin >> t;
while (t--) {
solve();
}
return 0;
}