1. 程式人生 > >UVA1471 Defense Lines

UVA1471 Defense Lines

題目解析

詳細可以檢視lrj書。本題題解出自劉汝佳演算法競賽入門經典

最容易想到的演算法是列舉j和i(前提是A[j]<A[i],否則拼不起來),然後分別往左和往右數一數最遠能延伸到哪裡。列舉量為O(n 2 ),而“數一數”的時間複雜度為O(n),因此總時間複雜度為O(n^{3} )。加上一個預處理,就能避免“數一數”這個過程,從而把時間複雜度降為O(n^{2} )。設right(i)為以第i個元素開頭的最長L序列長度,left(i)為以第i個元素結尾的最長L序列長度,則不難在O(n)時間內求出left(i)和right(i),然後列舉完j和i之後,最長L序列的長度就是left(j)+right(i)。還可以做得更好:只列舉i,不列舉j,而是用其他方法快速找一個j<i,使得A[j]<A[i],且left(j)儘量大。如何快速找到呢?首先要排除一些肯定不是最優值的j。例如,若有j'滿足A[j']<=A[j]且left(j')>left(j),則j肯定不滿足條件,因為j'不僅是一個更長的L序列的末尾,而且它更容易拼成。
這樣,把所有“有保留價值”的j按照A[j]從小到大排成一個有序表(根據剛才的結論,A[j]相同的j只保留一個),則g也會是從小到大排列。那麼用二分查詢找到滿足A[j]<A[i]的最大的A[j],則它對應的left(j)也是最大的。不過這個方法只有當i固定時才有效。實際上每次計算完一個g(i)之後,還要把這個A[i]加到上述有序表中,並且刪除不可能是最優的A[j]。因為這個有序表會動態變化,無法使用排序加二分查詢的辦法,而只能使用特殊的資料結構來滿足要求。幸運的是,STL中的set就滿足這個要求——set中的元素可以看成是排好序的,而且自帶lower_bound和upper_bound函式,作用和之前討論過的一樣。為了方便起見,此處用二元組(A[j],g(j))表示這些“有保留價值”的東西,如(10,4), (20,8),(30,15),  (40,18),  (50,30),並且以A[j]為關鍵字放在一個STL集合中。對於固定的i,不難用Lower_bound找到滿足A[j]<A[i]的最大A[j],以及對應的g(j),真正複雜的是這個集合本身的更新,即前面提到的“每次計算完一個g(i)之後”需要做的事情。假設已經計算出一個g(i)=6,且A[i]=25,接下來會發生什麼事情?首先把(25,6)插入集合中,然後檢查它的前一個元素(20,8)。由於20<25,8>6,(25,6)是不應該保留的。但如果插入的是(25,20),情況就完全不同了:不僅(25,20)需要保留,而且還要刪除(30,15)和(40,18)。一般地,插入任何一個二元組時首先應找到其插入位置,根據它前一個元素判斷是否需要保留。如果需要保留,再往後遍歷,刪除所有不再需要保留的元素。因為所有元素至多被刪除一次,而查詢、插入和刪除的時間複雜度均為O(logn),所以消耗在STL集合上的總時間複雜度為O(n\log n) 。

不過博主發現,本題不算很嚴謹,不含重複的序列的程式碼也能通過,至於含重複的序列的情況,博主暫未寫題解。先貼出不嚴謹的程式碼,以後再來補漏。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<set>
using namespace std;
const int maxn = 200000 + 5;
int n, a[maxn], left[maxn], right[maxn];
struct Candidate{
	int a, left;
	Candidate(int a, int left):a(a), left(left){}
	bool operator < (const Candidate &rhs) const {
		return a < rhs.a;
	}
};
set<Candidate> s;
int main(){
	int t, n;
	scanf("%d", &t);
	while(t--){
		scanf("%d", &n);
		for(int i = 0; i < n; ++i)
			scanf("%d", &a[i]);
		left[0] = 1;
		for(int i = 1; i < n; ++i) 
			if(a[i] > a[i - 1]){
				left[i] = left[i - 1] + 1;
			}else{
				left[i] = 1;
			}
		right[n - 1] = 1;
		for(int i = n - 2; i >= 0; --i)
			if(a[i] < a[i + 1]){
				right[i] = right[i + 1] + 1;
			}else{
				right[i] = 1;
			}
		s.clear();
		s.insert(Candidate(a[0], left[0]));
		int ans = 1;
		for(int i = 1; i < n; ++i){
			Candidate c(a[i], left[i]);
			set<Candidate>::iterator it = s.lower_bound(c);
			bool keep = true;
			if(it != s.begin()){
				Candidate last = *(--it);
				int len = last.left + right[i];
				ans = max(ans, len);
				if(last.left >= c.left) keep = false;
			}
			if(keep){
				s.insert(c);
				it = s.find(c);
				++it;
				while(it != s.end() && it->left <= c.left) s.erase(it++); 
			}
		}
		printf("%d\n", ans);
	}
	return 0;
}