線段樹專題—等差子序列 BZOJ-2124
阿新 • • 發佈:2018-12-13
線段樹專題—等差子序列
感謝
感謝孫耀峰的線段樹PPT,使我獲益匪淺.
題目來源
題意
給出長度為的的排列 問是否存在一組 使得構成等差序列 資料範圍
題解
只要能形成長度為的等差子序列,就直接輸出即可.
即需要尋找滿足
根據轉化後的式子,我們只需要列舉,判斷是否存在這樣的即可.
考慮到是一個排列,排列中的數兩兩不同,因此我們可以考慮開一個數據結構維護一下內容:
當列舉到的時候,我可以很快的查詢任何一個在的左邊還是右邊,(記如果則應該標記為表示在左邊,若則應該標記為表示在右邊).
這樣的話,如果串與之間有一位不相同(也就是說存在關於的對稱位置的兩個數在兩側,說明可以形成等差子序列).
因此只有當以上兩個01串,翻轉過來完全相同的情況下,才不存在以作為中心的等差序列.
如何判斷串相同呢?答案是雜湊.
建立一顆線段樹,維護的是串的正向和反向雜湊值.
從小到大列舉,然後更新線段樹,並判斷是否有以為對稱中心的等差序列.
程式碼
#include <iostream>
#include <cstring>
const int N = 10007;
typedef unsigned long long ull;
int T,n;
int a[N];
ull pow[N];
struct work{
ull hash0,hash1;
int len;
};
#define pr(x) std::cout << #x << ": " << x << std::endl
struct segtree{
ull hashs[N * 4][2];
void maintain(int rt,int l,int r) {
int mid = (l + r) / 2;
hashs[rt][0] = hashs[rt*2][0] + hashs[rt*2+1][0] * pow[mid - l + 1];
hashs[rt][1] = hashs[rt*2+1][1] + hashs[rt*2][1] * pow[r - mid];
}
void change(int rt,int l,int r,int pos,int val) {
if(l == r) {
hashs[rt][0] = hashs[rt][1] = val;
return ;
}
int mid = (l + r) / 2;
if(pos > mid)
change(rt*2+1,mid+1,r,pos,val);
else
change(rt*2,l,mid,pos,val);
maintain(rt,l,r);
}
work query(int rt,int l,int r,int ql,int qr) {
if(r < ql || qr < l) {
return (work){0,0,0};
}
else if(ql <= l && r <= qr) {
return (work){hashs[rt][0],hashs[rt][1],r - l + 1};
}
int mid = (l + r) / 2;
work wa = query(rt*2,l,mid,ql,qr);
work wb = query(rt*2+1,mid+1,r,ql,qr);
return (work){wa.hash0 + wb.hash0 * pow[wa.len],
wb.hash1 + wa.hash1 * pow[wb.len],
wa.len + wb.len};
}
}seg;
int main()
{
pow[0] = 1;
for(int i = 1;i < N;++i) pow[i] = 3 * pow[i-1];
std::ios::sync_with_stdio(false);
std::cin >> T;
while(T--) {
int f = 0;
memset(&seg,0,sizeof(seg));
std::cin >> n;
for(int i = 1;i <= n;++i) {
std::cin >> a[i];
seg.change(1,1,n,a[i],1);
if(a[i] == 1 || a[i] == n) continue;
int len = std::min(a[i]-1,n - a[i]);
work wa = seg.query(1,1,n,a[i]-len,a[i]-1);
work wb = seg.query(1,1,n,a[i]+1,a[i]+len);
if(wa.hash0 != wb.hash1) {
f = 1;
}
}
if(!f)
std::cout << "N" << std::endl;
else
std::cout << "Y" << std::endl;
}
return 0;
}