Codeforces Round #709 (Div. 2) D.Playlist

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;
}

上一篇:算法打卡Week4


下一篇:B. Navigation System(思维+反向建边最短路/最短路计数)