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];
    }
};