Problem Description
给你T,代表有T组测试数据,每组测试数据给你三个数n,s,m,n代表有n个结点1-n。s代表输出的起始点。m:代表两点之间的边权等于两点的编号相减的绝对值对m取模。让你输出序列,以s开始的序列。该序列满足距离现有的点,最大的边权点进来,如果边权一样,编号最小的点进来。
思路:
建边,因为1-n个点,而且是完全二叉树的结构。我们可以根据root的儿子是root<<1 || root<<1|1来建边。跑一遍dfs就可以把边建好了。然后用优先队列bfs,就可以直接处理上述问题。
#include<bits/stdc++.h>
using namespace std;
struct node
{
int to, w, next;
bool operator < (const node &b) const {
if(w == b.w) return to > b.to;
else return w < b.w;
}
};
#define lson root<<1
#define rson root<<1|1
node Map[1000055];
int cnt, head[500055], n, m, s, vis[500055];
void add(int u, int v, int w, int &cnt)//前向星存边
{
Map[cnt].to = v;
Map[cnt].w = w;
Map[cnt].next = head[u];
head[u] = cnt++;
}
void dfs(int root)
{
if((lson) <= n)//满足要求,建立双向边
{
add(root, lson, ((lson) - root) % m, cnt);
add(lson, root, ((lson) - root) % m, cnt);
}
else return;
if((rson) <= n)//满足要求,建立双向边
{
add(root, rson, ((rson) - root) % m, cnt);
add(rson, root, ((rson) - root) % m, cnt);
}
else return;
dfs(lson);
dfs(rson);
}
int main()
{
int T;
scanf("%d", &T);
while(T--)
{
scanf("%d %d %d", &n, &s, &m);
memset(head, -1, sizeof(head));
int root = 1;
cnt = 0;
dfs(root);//建边
priority_queue<node> q;//优先出来权值大的,否则出来编号小的
q.push((node){s, 0});
memset(vis, 0, sizeof(vis));
vector<int> ans;
while(!q.empty())//bfs
{
node t = q.top(); q.pop();
vis[t.to] = 1;
ans.push_back(t.to);//记录路径
for(int i = head[t.to]; ~i; i = Map[i].next)
{
int to = Map[i].to, w = Map[i].w;
if(!vis[to])
{
q.push((node){to, w});
}
}
}
for(int i = 0; i < ans.size(); i++)//输出路径
{
if(i) printf(" ");
printf("%d", ans[i]);
}
printf("\n");
}
}