打表
从左往右遍历记录每个index上左边最大的值,从右往左遍历记录每个index上右边最小的值,最后从左往右遍历利用打好的表即可。
#include<iostream>
#include<vector>
using namespace std;
int main(){
int n,count = 0;
cin>>n;
vector<int> leftMax(n),rightMin(n),a(n),ans(n);
for(int i = 0;i<n;i++){
cin>>a[i];
}
int ma = -1;
for(int i = 0;i<n;i++){
leftMax[i] = ma;
if(a[i]>ma) ma = a[i];
}
int mi = 1e10;
for(int i = n-1;i>=0;i--){
rightMin[i] = mi;
if(a[i]<mi) mi = a[i];
}
for(int i = 0;i<n;i++){
if(a[i]>=leftMax[i]&&a[i]<=rightMin[i]) ans[count++] = a[i];
}
cout<<count<<endl;
for(int i = 0;i<count;i++){
if(i!=0) cout<<" ";
cout<<ans[i];
}
return 0;
}
活用递推
就找关系,A左边的P乘以右边的T的数目即为这个A的PAT数目,也可以打表记录P,T的数量。
#include<iostream>
#include<string>
using namespace std;
int main(){
string query;
cin>>query;
int index = 0;
int ans = 0;
int countp = 0;
int countt = 0;
for(int i = 0;i<query.length();i++){
if(query[i]=='P'){
countp++;
}else if(query[i]=='A'){
index = i;
break;
}
}
for(int i = index+1;i<query.length();i++)
if(query[i]=='T')
countt++;
ans = (ans + countp*countt) % 1000000007;
for(int i = index+1;i<query.length();i++){
if(query[i]=='P'){
countp++;
}else if(query[i]=='T'){
countt--;
}else{
ans = (ans+countp*countt)%1000000007;
}
}
cout<<ans;
return 0;
}
随机选择算法
利用randPartition方法来找到无序序列中第k大的数,时间复杂度O(n),证明还不知道
//注意k值和m的比较
int randSelect(int a[],int left,int right,int k){
if(left == right) return a[left];
int randP = randPartition(a,left,right);
int m = randP - left + 1;
if(m == k) return a[randP] ;
if(k < m)
return randSelect(a,left,randP-1,k) ;
return randSelect(a,randP+1,right,k-m);
}