题目连接:Codeforces 416D Population Size
题目大意:给出一个序列,求最少可以划分成几个子的等差序列,-1代表任意值,但是一定要大于0.
解题思路:贪心,首先明确如果当前的数能够合进前一个序列,那么绝对不会让它留到下一个序列去考虑。
遍历,对于每个数进行判断,看是否可以并到前一个等差序列。首先先找到最前两个不为任意值的数,判断是否可以组成等差数列(因为会有-1干扰, -1 1 2,ans = 2),判断是否能组成,则要记录第一个数前面有几个-1,确定公差后会不会导致前面任意值变成负数。然后确定公差之后,只需要记录公差和前一项的值即可,如果新的数不能并入序列,则新开序列,按照上面的方法重新计算。
#include <stdio.h> #include <string.h> #include <iostream> using namespace std; typedef long long ll; ll n, a, pl, pr, val, cnt, ans, d, cur; void set(ll pos, int value, int count) { pl = pos; val = value; cnt = count; } int main () { pl = pr = -1; ans = cnt = 0; cin >> n; for (ll i = 1; i <= n; i++) { cin >> a; if (a < 0) { if (pl < 0) cnt++; else if (pr < 0) continue; else if (cur + d > 0) cur = cur + d; else { set(-1, 0, 1); pr = -1; ans++; } } else { if (pl < 0) { set(i, a, cnt); } else if (pr < 0) { ll sum = a - val; d = sum / (i - pl); if (val + (i-pl)*d != a || val <= cnt * d) { set(i, a, 0); ans++; } else { pr = i; cur = a; } } else { if (cur + d == a) cur = a; else { set(i, a, 0); pr = -1; ans++; } } } } cout << ans + 1<< endl; return 0; }