技能大賽卡題少女

遍歷圖一棵有根節(jié)點(diǎn)的樹,到達(dá)葉子節(jié)點(diǎn)不能遇到連續(xù)超過m個(gè)1,就是一個(gè)圖的遍歷,我TM 卡了這么久,是因?yàn)?葉子節(jié)點(diǎn)判斷錯(cuò)了,阿媽媽咪呀,我發(fā)現(xiàn) dfs 錯(cuò)也不容易查出來呀,以后要 多用printf 輸出檢查一下了。
鄰接表過的

#include <bits/stdc++.h>
using namespace std;
#define N 200005
#define ll long long
ll num[N];
int n,m;
struct Node{
int v,u;
}node[N];
int head[N];
int cnt=0;
int a,b;
int ans=0;
bool vis[N];
void init()
{
memset(head,-1,sizeof(head));
memset(vis,0,sizeof(vis));
cnt=0;
}
void add_e(int u,int v)
{
    node[cnt].v=v;
    node[cnt].u=head[u];
    head[u]=cnt++;
}
int dfs(int j,int cc)
{
vis[j]=true;
if(cc<=m)
{
    int f=0;
    for(int i=head[j];i!=-1;i=node[i].u)
    {
        if(!vis[node[i].v])
        {

                    if(num[node[i].v]!=1)
                    {

                        dfs(node[i].v,0);
                    }
                    else{
                        dfs(node[i].v,cc+1);
                    }
f=1;
        }
    }
     if(f==0){
                        //cout<<j<<endl;
                    if(cc<=m)
                    {
                     ans++;
                    }
                }
}
}
int main()
{
   // freopen("1.in","r",stdin);

    cin >> n >> m;
    init();
    for(int i=1;i<=n;i++)
    {
        cin >> num[i];
    }
    for(int i=0;i<n-1;i++)
    {
            cin >> a >> b;
           add_e(b,a);
           add_e(a,b);
    }
    if(num[1]!=1)
    {
     dfs(1,0);
    }
    else
    {
     dfs(1,1);
    }
    cout<<ans;
    return 0;
}

之前寫的代碼 ,鄰接表數(shù)組也沒乘2倍,cc 在葉子節(jié)點(diǎn)也改變了,但是沒調(diào)出來,(╥╯^╰╥)

#include <bits/stdc++.h>
using namespace std;
#define N 200005
#define ll long long
ll num[N];
int n,m;
struct Node{
int v,u;
}node[N];
int head[N];
int cnt=0;
int a,b;
int ans=0;
bool vis[N];
void init()
{
memset(head,-1,sizeof(head));
memset(vis,0,sizeof(vis));
cnt=0;
}
void add_e(int u,int v)
{
    node[cnt].v=v;
    node[cnt].u=head[u];
    head[u]=cnt++;
}
int dfs(int j,int cc)
{
vis[j]=true;
if(cc<=m)
{
    for(int i=head[j];i!=-1;i=node[i].u)
    {
        if(!vis[node[i].v])
        {
             int f=0;
                for(int k=head[node[i].v];k!=-1;k=node[k].u)
                {
                    if(vis[node[k].v]==0)
                    {
                        f=1;
                    }
                }
                if(f==0)
                {
                    int tc=cc;
               // cout<<node[i].v<<" "<<m<<" num "<<num[node[i].v]<<" c "<<cc<<endl;
                    if(num[node[i].v]==1)
                    {
                     tc++;
                    }
                    if(tc<=m)
                    {
                     ans++;
                     vis[node[i].v]=1;
                    }
                }
                else
                {
                    if(num[node[i].v]!=1)
                    {
                        dfs(node[i].v,0);
                    }
                    else{
                        dfs(node[i].v,cc+1);
                    }
                }
        }
    }
}
}
int main()
{
   // freopen("input.txt","r",stdin);
    cin >> n >> m;
    init();
    for(int i=1;i<=n;i++)
    {
        cin >> num[i];
    }
    for(int i=0;i<n-1;i++)
    {
           cin >> a >> b;
           add_e(b,a);
           add_e(a,b);
    }
    if(num[1]!=1)
    {
     dfs(1,0);
    }
    else
    {
     dfs(1,1);
    }
    cout << ans;
    return 0;
}

vector 過的

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define maxn 100003
vector<int>v[maxn];
int vis[maxn];
bool bv[maxn];
int ans=0;
int n,m;
void dfs(int no,int cn)
{
    bv[no]=1;
    if(vis[no]==1)
    {
        cn++;
    }
    else{
       cn=0;
    }
    if(cn<=m)
    {
        int f=0;
        for(int i=0;i<v[no].size();i++)
        {
            if(bv[v[no][i]]==0)
            {
             dfs(v[no][i],cn);
             f=1;
            }
        }
        if(f==0)
        {
                ans++;
        }

    }
}
int main()
{
 //freopen("input.txt","r",stdin);
 //freopen("output.txt","w",stdout);
    int t;
    int a,b;
    while(cin>>n>>m)
    {
        ans=0;
        memset(bv,0,sizeof(bv));
        t=0;
        for(int i=0;i<n;i++)
        {
            cin>>vis[i+1];
        }
        for(int i=0;i<n-1;i++)
        {
          cin>>a>>b;
          v[a].push_back(b);
          v[b].push_back(a);
        }
        dfs(1,0);
        printf("%d\n",ans);
    }
    return 0;
}

以后水題沒做出來,就重新寫一遍吧

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

推薦閱讀更多精彩內(nèi)容

  • 樹形動(dòng)態(tài)規(guī)劃,顧名思義就是樹+DP,先分別回顧一下基本內(nèi)容吧:動(dòng)態(tài)規(guī)劃:?jiǎn)栴}可以分解成若干相互聯(lián)系的階段,在每一個(gè)...
    Mr_chong閱讀 1,505評(píng)論 0 2
  • 1 概述 圖是數(shù)據(jù)結(jié)構(gòu)中最復(fù)雜的形式,也是最燒腦的結(jié)構(gòu)。無數(shù)的牛人樂此不疲地鉆研,然而,時(shí)至今日,依然有很多問題等...
    CodingTech閱讀 2,384評(píng)論 0 8
  • VisuAlgo!一,Date Structure的核心技術(shù)是分解和抽象二,基本概念和常用術(shù)語 三,邏輯結(jié)構(gòu)1,邏...
    斜杠青年許晏銘閱讀 918評(píng)論 0 0
  • 天氣進(jìn)入變態(tài)熱模式,我一整天躲在空調(diào)房里。下午突然想到,這么熱的天氣,會(huì)催生什么呢? 首先就是避暑的需求。趨利避害...
    黃曉折閱讀 226評(píng)論 0 0
  • 他是才子,是詞人,亦是帝王。這位帝王的失敗成就了他文壇旗幟般的地位。如果沒有作為亡國(guó)的經(jīng)歷,如果他沒有過人的文采,...
    山樵一閱讀 641評(píng)論 0 0