牛客练习赛24

A 石子阵列

题目链接：https://www.nowcoder.com/acm/contest/157/A

刚开始没看到相邻的石子不同这个条件，开心地直接套了Polya数的板子（由于昨天刚刚做过，记忆犹新，太激动了，其实这个也不是环），WA了一发。是一道水题，因为相邻的石子不一样，直接乘法原理。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cmath>
#define LL long long
using namespace std;
const int mod=1000000007;

int main()
{
    int n,m;
    while(cin>>n>>m)
    {
        LL ans=1;
        for(int i=0;i<n;i++)
        {
            if(i)
                ans=ans*(m-1)%mod;
            else
                ans=(ans*m)%mod;
        }
        cout<<ans<<endl;
    }
    return 0;
}

B 凤凰

题目链接：https://www.nowcoder.com/acm/contest/157/B

求出与1相连的子树节点数就行了，开始dfs TLE了一发，改成bfs就过了

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <vector>
#include <queue>
#define LL long long
using namespace std;
vector<int>v[1100000];
bool vis[1100000];
int dfs(int now)
{
    int cnt=0;
    vis[now]=1;
    int len=v[now].size();
    int next;
    for(int i=0;i<len;i++)
    {
        next=v[now][i];
        if(vis[next])
        continue;
        cnt++;
        cnt+=dfs(next);
    }
    return cnt;
}
int bfs(int x)
{
    queue<int> q;
    q.push(x);
    int cnt=0;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        int len=v[u].size();
        for(int i=0;i<len;i++)
        {
            int uu=v[u][i];
            if(!vis[uu])
            {
                cnt++;
                vis[uu]=1;
                q.push(uu);
            }
        }
    }
    return cnt;
}
int main()
{
    int n;
    scanf("%d",&n)    ;
        int a,b;
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            //cin>>a>>b;
            v[a].push_back(b);
            v[b].push_back(a);
        }
        vis[1]=1;
        int ans=0;

        for(int i=0;i<v[1].size();i++)
        {
            vis[v[1][i]]=1;
            ans=max(ans,bfs(v[1][i])+1);
        }
        printf("%d\n",ans);
        //cout<<ans<<endl;

    return 0;
}

C PH试纸

题目链接：https://www.nowcoder.com/acm/contest/157/C

刚开始看以为是线段树，后来发现直接暴力做就行

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cmath>
#define LL long long
using namespace std;
int red[1100000],blue[1100000];
 
int main()
{
    int n,m;
    while(cin>>n>>m)
    {
        //memset(red,0,siz)
        string s;
        cin>>s;
        int red_cnt=0,blue_cnt=0;
        for(int i=0;i<n;i++)
        {
            if(s[i]=='R')
                red[++red_cnt]=i+1;
            else
                blue[++blue_cnt]=i+1;
        }
        char c;int q;
        while(m--)
        {
            cin>>c>>q;
            if(c=='R')
            {
                if(red_cnt<q)
                    cout<<-1<<endl;
                else
                    cout<<red[q]<<endl;
            }
            else
            {
                if(blue_cnt<q)
                    cout<<-1<<endl;
                else
                    cout<<blue[q]<<endl;
            }
        }
    }
    return 0;
}

D 插排树

题目链接：https://www.nowcoder.com/acm/contest/157/D

求根节点到叶子节点的最长距离，dfs

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <vector>
#include <algorithm>
#define maxn 50005
using namespace std;
 
vector<long long> ans;
struct node
{
    int v,len;
    node(){}
    node(int a,int b){v=a,len=b;}
};
vector<node> G[maxn];
int vis[maxn];
int is_root[maxn];
void dfs(int u,long long d,int fa)
{
    int len=G[u].size();
    if(len==0)
        ans.push_back(d);
    for(int i=0;i<len;i++)
    {
        int v=G[u][i].v;
            dfs(v,d+G[u][i].len,u);
    }
}
int main()
{
    int n,x,y,d;
//    freopen("out.txt","r",stdin);
    cin>>n;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d%d",&x,&y,&d);
        is_root[x]=1;
        G[y].push_back(node(x,d));
    }
    for(int i=1;i<=n;i++)
        if(!is_root[i])
    {
        dfs(i,0,-1);break;
    }
    sort(ans.begin(),ans.end(),greater<int>());
    cout<<ans[0]<<endl;
}
/*
9
2 1 2
3 1 2
4 1 1
5 2 3
6 2 1
7 3 1
8 3 4
9 7 5
10 5 99
11 6 5
12 8 3
13 14 100
13 15 101
13 4 2
*/

E 青蛙

题目链接：https://www.nowcoder.com/acm/contest/157/E

一开始就被dp的思路套住了，后来各种漏看条件，结束前3分钟，队友发现是最短路，然后疯狂找板子，在比赛结束后30sA掉

#include <bits/stdc++.h>
using namespace std;
//par[i][j]:i和j间距，不相连就是MAX
#define maxn 300
#define INF 0x3f3f3f3f
int vis[maxn];
int par[maxn][maxn];
int dis[maxn];
int n,m;
void Dij()
{
    memset(vis,0,sizeof(vis));
    for( int i=2; i<=n; i++ )
    dis[i] = par[1][i];
    dis[1] = 0;
    vis[1] = 1;
    for( int i=0; i<n; i++ )
    {
        int temp= INF;
        int v = -1; // 定义一个-1 ，方便判断是不是有边的存在（或边是否已全部加入 已连接点 的集合）
        for( int j=1; j<=n; j++ ) // 由于每一个点都要进行一遍边的判定，所以是两个for
        {
            if( temp > dis[j] && !vis[j] )
            {
                temp = dis[j];
                v = j;
            }
        }
        if( v == -1 )  return; //如果全部边加完了（或者没有边），直接退出
        vis[v] = 1;
        for( int j=1; j<=n; j++ )// 更新
        {
            if( !vis[j] && dis[j] > dis[v]+par[v][j] )
                dis[j] = dis[v]+par[v][j];
        }
    }
}
int main()
{
    cin>>n>>m;
    for(int i=0;i<=n;i++)
        for(int j=0;j<=n;j++) par[i][j]=INF;
    for(int i=0;i<n;i++)
    {
        par[i][i+1]=1;
        par[i+1][i]=1;
    }
    int x,y;
    while(m--)
    {
        scanf("%d%d",&x,&y);
        par[x][y]=1;
    }
    Dij();
    cout<<dis[n]+1<<endl;
}
/*
16 4
2 10
8 15
12 5
13 6
*/