D题
题目链接
题意:给你一个长度为n序列,每个元素都是正整数,每次选择相邻的元素(x<y)进行操作(即gcd(x,y)==1),就删除y元素,例如[5, 9, 2, 10, 15] → [5, 9, 2, 10, 15] → [5, 2, 10, 15] (因为 gcd(5,9)=1) → [5, 2, 10, 15] → [5, 2, 10, 15] → [5, 2, 10, 15] → [5, 2, 10, 15] → [5, 2, 10, 15] → [5, 10, 15] (因为 gcd(5,2)=1) → [5, 10, 15] → [5, 10, 15] → … 。求删除元素的个数和相应的下标。
题解
用双端队列进行模拟,不懂的可以在草稿纸上手动的走几遍就懂了。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=1e5+5;
int arr[MAXN],nxt[MAXN],vis[MAXN],ans[MAXN];
int main()
{
int t;cin>>t;
while(t--){
int n;cin>>n;
deque<int>que;
for(int i=1;i<=n;i++){
cin>>arr[i];
nxt[i]=i+1;
vis[i]=0;
que.push_back(i);
}
int cnt=0;
nxt[n]=1;
while(!que.empty()){
int x=que.front();que.pop_front();
if(vis[x])continue;
if(__gcd(arr[x],arr[nxt[x]])==1){
ans[++cnt]=nxt[x];
vis[nxt[x]]=1;
nxt[x]=nxt[nxt[x]];
que.push_back(x);
}
}
cout<<cnt<<" ";
for(int i=1;i<=cnt;i++)cout<<ans[i]<<" ";
cout<<endl;
}
return 0;
}