CCF-CSP 202112-5 极差路径暴力12分

最新推荐文章于 2022-09-25 22:03:43 发布

自信的小螺丝钉

最新推荐文章于 2022-09-25 22:03:43 发布

阅读量459

点赞数 1

分类专栏： CCF-CSP 文章标签：深度优先图论算法

本文链接：https://blog.csdn.net/qq_45791939/article/details/125172115

版权

CCF-CSP 专栏收录该内容

80 篇文章 5 订阅

订阅专栏

这是一个关于解决 CCF CSP（中国计算机学会）竞赛中的极差路径问题的程序。程序通过深度优先搜索遍历图中节点，寻找在给定范围内包含两个节点的极差路径，并计算符合条件的路径数量。输入包括节点数、范围参数以及图的连接信息，输出为符合条件的路径数。

摘要由CSDN通过智能技术生成

原题链接：CCF-CSP 202112-5 极差路径
在这里插入图片描述

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int N=5e5+10;

int n,k1,k2;
vector<int> g[N];
vector<int> v;
int vis[N];
set<pair<int,int>> res;

void dfs(int x,vector<int>& v,int y)
{
    int minnum=min(x,y)-k1;
    int maxnum=max(x,y)+k2;
    int minp=*(min_element(v.begin(),v.end()));
    int maxp=*(max_element(v.begin(),v.end()));
    if(minnum<=minp && minp<=maxp && maxp<=maxnum)
    {
        int a=x,b=y;
        if(a>b) swap(a,b);
        pair<int,int> p=make_pair(a,b);
        res.insert(p);
    }
    for(int i=0;i<g[y].size();i++)
    {
        int tmp=g[y][i];
        if(vis[tmp]==0)
        {
            v.push_back(tmp);
            vis[tmp]=1;
            dfs(x,v,tmp);
            v.pop_back();
            vis[tmp]=0;
        }
    }
}

int main()
{
    std::ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    cin>>n>>k1>>k2;
    if(n<=5000)
    {
        for(int i=0;i<n-1;i++)
        {
            int x,y;
            cin>>x>>y;
            g[x].push_back(y);
            g[y].push_back(x);
        }
        for(int i=1;i<=n;i++)
        {
           memset(vis,0,sizeof(vis));
           v.clear();
           v.push_back(i);
           vis[i]=1;
           dfs(i,v,i);
        }
        cout<<res.size()<<endl;
    }
    return 0;
}