結果

問題 No.826 連絡網
ユーザー S97323424
提出日時 2020-10-11 12:25:15
言語 Lua
(LuaJit 2.1.1734355927)
結果
AC  
実行時間 1,552 ms / 2,000 ms
コード長 1,283 bytes
コンパイル時間 43 ms
コンパイル使用メモリ 5,248 KB
実行使用メモリ 35,840 KB
最終ジャッジ日時 2024-07-20 17:14:40
合計ジャッジ時間 13,636 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

local UnionFind={}

UnionFind.create=function(self,n)
    self.par={}
    self.rank={}
    self.size={}
    for i=1,n do
        self.par[i]=i
        self.rank[i]=1
        self.size[i]=1
    end
end

UnionFind.getroot=function(self,x)
    local nx=x
    while self.par[x]~=x do
        x=self.par[x]
    end
    while self.par[nx]~=x do
        self.par[nx],nx=x,self.par[nx]
    end
    return x
end

UnionFind.same=function(self,x,y)
    return self:getroot(x)==self:getroot(y)
end

UnionFind.unite=function(self,x,y)
    x=self:getroot(x)
    y=self:getroot(y)
    if x==y then
        return
    end
    if self.rank[x]>self.rank[y] then
        x,y=y,x
    elseif self.rank[x]==self.rank[y] then
        self.rank[x]=self.rank[x]+1
    end
    self.par[x]=y
    self.size[y]=self.size[y]+self.size[x]
end

UnionFind.getsize=function(self,x)
    return self.size[self:getroot(x)]
end

UnionFind.new=function(self,n)
    local obj={}
    setmetatable(obj,{__index=UnionFind})
    obj:create(n)
    return obj
end

----------

local n,p=io.read("*n","*n")
local uf=UnionFind:new(n)
local t={}
for i=1,n do
    t[i]=0
end
for i=2,n do
    if 1>t[i] then
        for j=2,math.floor(n/i) do
            t[i]=1
            uf:unite(i,i*j)
        end
    end
end
print(uf:getsize(p))
0