1. 程式人生 > >初賽A輪

初賽A輪

急了 ras return 兩個 但是 [1] div blog pre

數列互質

時間限制:6秒

空間限制:524288K

給出一個長度為 n 的數列 { a[1] , a[2] , a[3] , ... , a[n] },以及 m 組詢問 ( l[i] , r[i] , k[i])。
求數列下標區間在 [ l[i] , r[i] ] 中有多少數在該區間中的出現次數與 k[i] 互質(最大公約數為1)。
輸入描述:
第一行,兩個正整數 n , m (1 ≤ n, m ≤ 50000)。
第二行,n 個正整數 a[i] (1 ≤ a[i] ≤ n)描述這個數列。
接下來 m 行,每行三個正整數 l[i] , r[i] , k[i] (1 ≤ l[i] ≤ r[i] ≤ n, 1 ≤ k[i] ≤ n),描述一次詢問。


輸出描述:
輸出 m 行,即每次詢問的答案。

輸入例子:
10 5
1 1 1 1 1 2 2 2 2 2
4 7 2
4 7 3
4 8 2
4 8 3
3 8 3

輸出例子:
0
2
1
1
0

思路:A輪2題。。這題知道是莫隊+數據結構,但是沒學過莫隊,就去學習了一發,然後最後半個小時太心急了,導致沒拍完,。。我的T恤啊。。啊。。(另外寫了的2題就不補了,心疼,莫隊的板子套的是HZWER大佬的,謝謝大佬)

#include<bits/stdc++.h>
using namespace std;

const int N=50004;
int a[N];
set<int >s;
set<int >::iterator it; int ma[N]; struct node{ int l,r,x,xx; int block; }e[N]; int c[N]; bool cmp(node p,node q){ if(p.block==q.block) return p.r<q.r; return p.block<q.block; } int gcd(int a,int b) { if(b==0) return a; return gcd(b,a%b); } int main(){
int n,m; scanf("%d%d",&n,&m); int size=sqrt(n); for(int i=1;i<=n;i++) scanf("%d",&a[i]); for(int i=1;i<=m;i++){ scanf("%d%d%d",&e[i].l,&e[i].r,&e[i].x); e[i].xx=i; e[i].block=(e[i].l-1)/size+1; } sort(e+1,e+1+m,cmp); int l=1,r=0; for(int i=1;i<=m;i++){ int sum=0; while(l>e[i].l){l--;s.insert(a[l]);ma[a[l]]++;} while(r<e[i].r){r++;s.insert(a[r]);ma[a[r]]++;} while(l<e[i].l){ ma[a[l]]--;if(ma[a[l]]==0) s.erase(a[l]);l++;} while(r>e[i].r) { ma[a[r]]--;if(ma[a[r]]==0) s.erase(a[r]);r--;} for(it=s.begin();it!=s.end();it++){ if(gcd(ma[*it],e[i].x)==1){ sum++; } } c[e[i].xx]=sum; } for(int i=1;i<=m;i++)printf("%d\n",c[i]); return 0; }

初賽A輪