JZOJ 5048. 【GDOI2017模拟一试4.11】IQ测试
Description
PTY進(jìn)行IQ測試,測試的項目是判斷一個序列是否是另外一個序列刪除若干個數(shù)字之后得
到的,PTY 深知自己的IQ 低于sqrt(-1),所以他請來了智商超高的你來替他解決問題。
Input
第一行為一個整數(shù)n,第二行包括n個用空格分開的整數(shù)ai,組成了最初的序列,第三行為一個整數(shù)m,表示n個IQ測試詢問的序列,每個序列兩行,第一行給出長度 L(1<=L<=n),然后下一行為L 個由空格分開的整數(shù)bi。
Output
共m行,如果詢問的序列確實是由最初的序列刪除一些數(shù)得到,就輸出TAK,否則輸出NIE。
Sample Input
7
1 5 4 5 7 8 6
4
5
1 5 5 8 6
3
2 2 2
3
5 7 8
4
1 5 7 4
Sample Output
TAK
NIE
TAK
NIE
Data Constraint
對于30%的數(shù)據(jù):n<=1000,m<=1000
對于100%的數(shù)據(jù):1<=ai,bi<=1000000,ΣL<=1000000,n,m<=1000000
Solution
這題再一看很簡單,維護(hù)兩個指針即可,但是一看數(shù)據(jù)范圍就嚇個半死~
當(dāng)我靜下心來思考時,就發(fā)現(xiàn)沒有想象中的那么恐怖。
注意到 ∑L≤106 ,于是考慮逐個逐個數(shù)列處理。
當(dāng)處理到第i個數(shù)列時,將每一個元素在原數(shù)列中查找,當(dāng)查找不到時則輸出“NIE”;
怎樣查找呢?——可以 O(logN) 查找。
開一個Vector,讀入原數(shù)列時,將當(dāng)前值所處的的位置 Push進(jìn)當(dāng)前值的Vector中,
查找即可在Vector中二分查找未匹配的位置中最前的那一個,之后更新已匹配位置即可。
這樣的時間復(fù)雜度為 NlogN 級別,加上 讀入輸出優(yōu)化 可以避免卡常!
Code
#include<cstdio> #include<vector> #define P(x) putchar(x) using namespace std; vector<int>f[1000001]; inline int read() {int data=0; char ch=0;while(ch<'0' || ch>'9') ch=getchar();while(ch>='0' && ch<='9') data=data*10+ch-'0',ch=getchar();return data; } int main() {int n=read();for(int i=1;i<=n;i++) f[read()].push_back(i);int m=read();while(m--){int k=read(),last=0;bool bz=true;while(k--){int x=read();int l=0,r=f[x].size()-1;if(r<0 || f[x][r]<=last){bz=false;while(k--) read();break;}while(l<r){int mid=(l+r)>>1;if(f[x][mid]<=last) l=mid+1; else r=mid;}if(f[x][l]>last) last=f[x][l]; else{bz=false;while(k--) read();break;}}if(bz) P('T'),P('A'),P('K'); else P('N'),P('I'),P('E');P('\n');}return 0; }總結(jié)
以上是生活随笔為你收集整理的JZOJ 5048. 【GDOI2017模拟一试4.11】IQ测试的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: JZOJ 2256. 【BZOJ 225
- 下一篇: JZOJ 5050. 【GDOI2017