1065 单身狗

单身狗

题目描述:

“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人,以便给予特殊关爱。

输入格式:

输入第一行给出一个正整数 N(≤ 50 000),是已知夫妻/伴侣的对数;随后 N 行,每行给出一对夫妻/伴侣——为方便起见,每人对应一个 ID 号,为 5 位数字(从 00000 到 99999),ID 间以空格分隔;之后给出一个正整数 M(≤ 10 000),为参加派对的总人数;随后一行给出这 M 位客人的 ID,以空格分隔。题目保证无人重婚或脚踩两条船。

输出格式:

首先第一行输出落单客人的总人数;随后第二行按 ID 递增顺序列出落单的客人。ID 间用 1 个空格分隔,行的首尾不得有多余空格。

输入样例:

3
11111 22222
33333 44444
55555 66666
7
55555 44444 10000 88888 22222 11111 23333

输出样例:

5
10000 23333 44444 55555 88888

思路:

代码:

//最后两个测试点会超时
package main

import (
    "fmt"
    "sort"
)

func main()  {
    var n int
    var s1,s2 string
    var id []string
    var flag [10001]int
    mp:= make(map[string]string,50001)
    fmt.Scan(&n)
    for i:=0;i<n;i++{
        fmt.Scan(&s1,&s2)
        if s1>s2{
            mp[s2]=s1
        }else {
            mp[s1]=s2
        }
    }
    fmt.Scan(&n)
    for i:=0;i<n;i++{
        fmt.Scan(&s1)
        id = append(id,s1)
    }
    sort.Strings(id)
    result:=n
    for i:=0;i<n;i++{
        if flag[i]==0&&mp[id[i]]!="0"{
            for k:=i+1;k<n;k++{
                if mp[id[i]]==id[k]{
                    flag[i]=1
                    flag[k]=1
                    result-=2
                    break
                }
            }
        }
    }
    fmt.Println(result)
    first:=0
    for i:=0;i<n;i++{
        if flag[i]==0{
            if first==0{
                fmt.Printf("%s",id[i])
                first=1
            }else{
                fmt.Printf(" %s",id[i])
            }
        }
    }
   
}

本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!