codeforces Playlist 循环链表 + BFS

题意 : 给定一个长度为n的数列, 首位相接成一个环, 不断遍历这个环, 若 gcd(a_i, a_i + 1) = 1则删除a_i + 1,并且i++, 直到所有相邻的数gcd均不为1

//https://codeforces.com/contest/1484/problem/D
#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;

const ll inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const double eps = 1e-4;
const ll mod = 1e9 + 7;
const ll N = 1e5 + 5;

ll n, a[N], nxt[N], vis[N];//nxt[i]记录i后第一个未被删除的数的下标
                           //vis[i]记录下标为i的数是否被删除, 若a[i] 没有被删除,则a[i+1]不可能被删除
ll gcd(ll x, ll y)
{
    return y ? gcd(y, x % y) : x;
}

void solve()
{
    cin >> n;
    for (ll i = 1; i <= n; i++)
        vis[i] = 0;
    for (ll i = 1; i <= n; i++)
        cin >> a[i];
    queue<ll> q;
    a[n + 1] = a[1];
    for (ll i = 1; i <= n; i++)
    {
        if (gcd(a[i], a[i + 1]) == 1)
            q.push(i);
        nxt[i] = i % n + 1;
    }
    vector<ll> ans;
    while (q.size())
    {
        ll now = q.front();
        q.pop();
        if (vis[now])//若下标为now的数被删除,则nxt[now]无需删除, 因为now删除后需i向后移1
            continue;
        vis[nxt[now]] = 1;
        ans.push_back(nxt[now]);
        nxt[now] = nxt[nxt[now]];//nxt的转移
        if (gcd(a[now], a[nxt[now]]) == 1)若删除后相邻的两个数gcd为1则扔进队列
            q.push(now);
    }
    cout << ans.size() << ' ';
    for (auto it : ans)
        cout << it << ' ';
    cout << '\n';
}

signed main()
{
    IOS;
    ll t = 1;
    cin >> t;
    while (t--)
        solve();
    return 0;
}

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值