cf264 C. Choosing Balls(dp)
阿新 • • 發佈:2022-01-08
題意:
n個球排成一列,每個球有值 \(v_i\) 和顏色 \(c_i\)。可任取子序列,對子序列中的某個球,若它不是子序列首且與子序列中的上一個球同色,則價值為 \(av_i\);若它是子序列首或者與子序列中的上一個球異色,則價值為 \(bv_i\)。子序列的價值定義為子序列中所有球的價值和。
現有q次詢問,每次給定 \(a,b\)。對每個詢問,找一價值最大的子序列並輸出價值。
\(1\le n \le 1e5, 1\le q \le 500, |v_i|\le 1e5,1\le c_i \le n, |a|\le 1e5,|b| \le 1e5\)
思路:
\(f(c)\) 表示最後一個球的顏色為 \(c\)
每個球 \(i\) 可以接到前面的末尾同色的子列後:\(f(c_i)=f(c_i) + av_i\)
也可以接到前面的最大的末尾異色子列後:\(f(c_i)=\mathop{max}\limits _{c_j\neq c_i}\{f(c_j)\} + bv_i\)
其中最大的 \(f\) 的值是遞增的,可以邊迴圈邊維護。為了異色,要維護最大顏色和次大顏色。
複雜度 \(O(nq)\)
int a, b, mi1 = 0, mi2 = 0; cin >> a >> b; memset(f, 0xcf, sizeof f); f[0] = 0; for(int i = 1; i <= n; i++) { int t = (c[i] != c[mi1]) ? mi1 : mi2; //異色 f[c[i]] = max({f[c[i]], f[c[i]] + (ll)v[i] * a, f[c[t]] + (ll)v[i] * b}); //dp if(f[c[mi1]] < f[c[mi2]]) swap(mi1, mi2); //維護最大和次大 if(c[i] == c[mi1] || c[i] == c[mi2]) continue; if(f[c[i]] > f[c[mi1]]) mi2 = mi1, mi1 = i; else if(f[c[i]] > f[c[mi2]]) mi2 = i; }