LeetCode 第179场周赛 题解

传送门
水一水真开心.jpg

A 生成每种字符都是奇数个的字符串

题目大意

生成一个长度为n的字符串，要求只能出现小写字母，并且任意字母的个数均为奇数个。

解题思路

若n为奇数，则输出奇数个相同字母即可
若n为偶数，则输出两个字母，每个字母都为奇数个即可

代码

class Solution {
public:
    string generateTheString(int n) {
        string ans="";
        if (n&1)
        {
            for (int i=0;i<n;++i)
                ans+='a';
        }
        else
        {
            ans+='a';
            for (int i=0;i<n-1;++i)
                ans+='b';
        }
        return ans;
    }
};

B 灯泡开关

题目大意

n个灯泡排成一排，一个灯泡变成蓝色当且仅当它左边的灯泡都亮着。
给出点亮灯泡的顺序，求有几个时刻亮着的灯泡都为蓝色。

解题思路

维护当前开了几个灯泡cnt，最右边的亮的灯泡Max，当cnt==Max的时候即为合法的时刻，统计即可。
正确性显然。

代码

class Solution {
public:
    int numTimesAllBlue(vector<int>& light) {
        int n=light.size();
        int Max=0;
        int cnt=0;
        int ans=0;
        for (int i=0;i<n;++i)
        {
            ++cnt;
            Max=max(Max,light[i]);
            if (cnt>=Max) ++ans;
        }
        return ans;
    }
};

C 通知所有员工所需的时间

题目大意

给出一棵树和每条边的长度，求树中的最长链。

解题思路

煞笔bfs

代码

class Solution {
public:
    int tot,point[100005],nxt[100005],v[100005],c[100005];
    queue <int> q;
    int dis[100005],vis[100005];
    
    void addedge(int x,int y,int z)
    {
        ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
    }
    int numOfMinutes(int n, int headID, vector<int>& manager, vector<int>& informTime) {
        for (int i=0;i<n;++i)
        {
            if (manager[i]!=-1)
                addedge(manager[i],i,informTime[manager[i]]);
        }
        memset(vis,0,sizeof(vis));
        memset(dis,0,sizeof(dis));
        while (!q.empty()) q.pop();
        q.push(headID);
        while (!q.empty())
        {
            int now=q.front();
            q.pop();
            for (int i=point[now];i;i=nxt[i])
                if (!vis[v[i]])
                {
                    vis[v[i]]=1;
                    dis[v[i]]=dis[now]+c[i];
                    q.push(v[i]);
                }
        }
        int ans=0;
        for (int i=0;i<n;++i)
            ans=max(ans,dis[i]);
        return ans;
    }
};

D T秒后青蛙的位置

题目大意

给出一个n个节点的树，青蛙从根开始跳，每一秒等概率地选择跳到某一个儿子上（到达叶子节点之后会一直原地不动）。问第t秒时青蛙跳到第target个节点上的概率。

解题思路

f(i,j)表示第i秒时跳到第j个节点上的概率。dp即可。

代码

class Solution {
public:
    double f[55][105];
    int tot,point[105],nxt[205],v[205];
    int size[105];
    double p[105];
    
    void addedge(int x,int y)
    {
        ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
        ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
    }
    void dfs(int x,int fa)
    {
        size[x]=0;
        for (int i=point[x];i;i=nxt[i])
            if (v[i]!=fa)
            {
                ++size[x];
                dfs(v[i],x);
            }
        if (size[x])
            p[x]=1/(double)size[x];
    }
    void dp(int x,int fa,int t)
    {
        if (!size[x])
        {
            f[t][x]+=f[t-1][x];
            return;
        }
        for (int i=point[x];i;i=nxt[i])
            if (v[i]!=fa)
            {
                f[t][v[i]]+=f[t-1][x]*p[x];
                dp(v[i],x,t);
            }
    }
    double frogPosition(int n, vector<vector<int>>& edges, int t, int target) {
        int m=edges.size();
        for (int i=0;i<m;++i)
            addedge(edges[i][0],edges[i][1]);
        dfs(1,0);
        memset(f,0,sizeof(f));
        f[0][1]=1.0;
        for (int i=1;i<=t;++i)
            dp(1,0,i);
        return f[t][target];
    }
};