【BFS && 树】UVALive - 7460 Maximum Cut Order

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