Init

ACMOJ-1665.cpp

@@ -0,0 +1,60 @@
+#include<iostream>
+#include<string>
+#include<algorithm>
+#include<cstring>
+using namespace std;
+const int maxn=2e4+5;
+int fa[maxn],cnt[maxn];
+int a[maxn],acnt;
+bool vis[maxn*2];
+int ff(int u)
+{
+    int a=u,b;
+    while(u!=fa[u]) u=fa[u];
+    while(a!=u)
+    {
+        b=fa[a];
+        fa[a]=u;
+        a=b;
+    }
+    return u;
+}
+int main()
+{
+#ifdef local
+	freopen("pro.in","r",stdin);
+#endif
+	int n,m,k; scanf("%d%d%d",&n,&m,&k);
+    for(int i=1;i<=n;i++) fa[i]=i;
+    for(int i=0;i<k;i++)
+    {
+        int u,v; scanf("%d%d",&u,&v);
+        fa[ff(u)]=ff(v);
+    }
+    for(int i=1;i<=n;i++)
+        cnt[ff(i)]++;
+    for(int i=1;i<=n;i++)
+        if(cnt[i])
+            a[acnt++]=cnt[i];
+    sort(a,a+acnt);
+    vis[0]=1;
+    for(int i=0;i<acnt;i++)
+        for(int j=n;j>=a[i];j--)
+            vis[j]|=vis[j-a[i]];
+    for(int i=0;;i++)
+    {
+        if(m-i<0&&m+i>n)
+            break;
+        if(m-i>=0&&vis[m-i])
+        {
+            printf("%d\n",m-i);
+            return 0;
+        }
+        if(m+i<=n&&vis[m+i])
+        {
+            printf("%d\n",m+i);
+            return 0;
+        }
+    }
+	return 0;
+}