{"id":837,"date":"2019-03-28T16:51:47","date_gmt":"2019-03-28T08:51:47","guid":{"rendered":"http:\/\/SmokeyDays.top\/wordpress\/?p=837"},"modified":"2019-03-28T16:52:22","modified_gmt":"2019-03-28T08:52:22","slug":"lp2664-%e6%a0%91%e4%b8%8a%e6%b8%b8%e6%88%8f","status":"publish","type":"post","link":"http:\/\/SmokeyDays.top\/wordpress\/2019\/03\/28\/lp2664-%e6%a0%91%e4%b8%8a%e6%b8%b8%e6%88%8f\/","title":{"rendered":"lp2664 \u6811\u4e0a\u6e38\u620f"},"content":{"rendered":"\n<p>\u4e00\u79cd\\(n^2log^2n\\)\u7684\u505a\u6cd5\u662f\u53ef\u4ee5\u60f3\u5230\u7684\uff1a\u70b9\u5206\u6cbb\uff0c\u4f7f\u7528multiset\u6765\u7ef4\u62a4\u4ee5\u67d0\u4e2a\u8282\u70b9\u4e3a\u6839\u7684\u67d0\u4e00\u4e9b\u94fe\u4e0a\u7684\u989c\u8272\u4fe1\u606f\u3002\u5408\u5e76\u7684\u65f6\u5019\\(O(nlogn)\\)\u5408\u5e76\u3002<br>\n\u6211\u4eec\u7565\u5fae\u8ba1\u7b97\u4e86\u4e00\u4e0b\u590d\u6742\u5ea6\uff0c\u611f\u89c9\u8fd9\u662f\u5361\u4e0d\u8fc7\u53bb\u7684\u3002<\/p>\n\n\n\n<p>\u4ed4\u7ec6\u60f3\u60f3\u6211\u4eec\u53d1\u73b0\u8ba1\u7b97\u4e00\u6761\u8def\u5f84\u4e0a\u4e0d\u540c\u7684\u989c\u8272\u6570\u91cf\u662f\u8f83\u4e3a\u56f0\u96be\u7684\uff0c\u6545\u800c\u6211\u4eec\u53ef\u4ee5\u8f6c\u800c\u7edf\u8ba1\u4e0d\u540c\u989c\u8272\u5bf9\\(ans_i\\)\u7684\u8d21\u732e\u3002<br>\n\u6211\u4eec\u4ed4\u7ec6\u601d\u8003\uff0c\u53d1\u73b0\uff0c\u5bf9\u4e8e\u4ee5\u4e00\u4e2a\u70b9\\(rt\\)\u4e3a\u6839\u7684\u5b50\u6811\u7684\u67d0\u4e2a\u70b9\\(i\\)\uff0c\u5176\u989c\u8272\u662f\u4ece\\(i\\)\u5230\\(rt\\)\u7684\u8def\u5f84\u4e0a\u9996\u6b21\u51fa\u73b0\u7684\uff0c\u90a3\u4e48\u6211\u4eec\u5728\u8ba1\u7b97\u4efb\u610f\u4e00\u4e2a\u5176\u4ed6\u5b50\u6811\u5185\u7684\u70b9\\(j\\)\u7b54\u6848\u65f6\uff0c\u53ef\u4ee5\u5c06\u7b54\u6848\u51cf\u53bb\\(sz_i\\)\uff0c\u7136\u540e\u518d\u5bb9\u65a5\u6389\\(j\\)\u5230\\(rt\\)\u8def\u5f84\u4e0a\u5b58\u5728\\(a_i\\)\u8fd9\u79cd\u989c\u8272\u7684\u60c5\u51b5\u3002<br>\n\u6545\u800c\uff0c\u5bf9\u4e8e\u6bcf\u4e00\u6b21\u5206\u6cbb\u8981\u5904\u7406\u7684\u4ee5\\(rt\\)\u4e3a\u6839\u7684\u6811\uff0c\u6211\u4eec\u53ef\u4ee5\u5148\u9884\u5904\u7406\u51fa\\(sm\\)\uff0c\u8868\u793a\u7684\u662f\u8fd9\u68f5\u6811\u4e2d\u6240\u6709\u6ee1\u8db3\u300c\u5230\u6839\u8282\u70b9\u8def\u5f84\u4e0a\u6ca1\u6709\u4e0e\u5176\u989c\u8272\u76f8\u540c\u7684\u70b9\u300d\u7684\u70b9\u7684\u5b50\u6811\u5927\u5c0f\u4e4b\u548c\uff1b\u4ee5\u53ca\\(val_i\\)\uff0c\u8868\u793a\u7684\u662f\u989c\u8272\u4e3a\\(i\\)\u7684\u5230\u6839\u8282\u70b9\u8def\u4e0a\u6ca1\u6709\u989c\u8272\u4e3a\\(i\\)\u7684\u8282\u70b9\u7684\u5b50\u6811\u5927\u5c0f\u4e4b\u548c\u3002\u7136\u540e\u6211\u4eec\u9700\u8981\u5bf9\u4ee5\u5b83\u7684\u6bcf\u4e00\u4e2a\u5b50\u8282\u70b9\u4e3a\u6839\u7684\u5b50\u6811\u7edf\u8ba1\u7b54\u6848\u3002<br>\n\u663e\u800c\u6613\u89c1\u7684\uff0c\\(j\\)\u5230\\(rt\\)\u7684\u8def\u5f84\u4e0a\u7ecf\u8fc7\u7684\u989c\u8272\u662f\u4e0d\u5e94\u8be5\u88ab\u91cd\u590d\u7edf\u8ba1\u7684\uff0c\u6240\u4ee5\u6211\u4eec\u8981\u8ba9\\(sm\\)\u51cf\u53bb\\(\\sum val_{i}\\)\uff0c\u5176\u4e2d\\(i\\)\u662f\u8def\u5f84\u4e0a\u7ecf\u8fc7\u7684\u989c\u8272\u3002<br>\n\u6b64\u5916\uff0c\u6211\u4eec\u8fd8\u9700\u8981\u7edf\u8ba1\u8fd9\u4e9b\u989c\u8272\u5bf9\u7b54\u6848\u7684\u8d21\u732e\u3002\u4ee4\u8fd9\u4e9b\u989c\u8272\u7684\u6570\u91cf\u4e3a\\(nm\\)\uff0c\u90a3\u4e48\u5b83\u4eec\u5bf9\\(j\\)\u7684\u7b54\u6848\u7684\u8d21\u732e\u5c31\u5e94\u8be5\u662f\\(nm*dlt\\)\uff0c\u5176\u4e2d\\(dlt\\)\u662f\\(rt\\)\u9664\u4e86\u5f53\u524d\u6b63\u5728\u627e\u7684\u5b50\u8282\u70b9\u4ee5\u5916\u7684\u5b50\u8282\u70b9\u7684\u5b50\u6811\u5927\u5c0f\u4e4b\u548c\u3002<br>\n\u7136\u540e\uff0c\u6211\u4eec\u8fd8\u9700\u8981\u7edf\u8ba1\u4ee5\u6839\u8282\u70b9\u4e3a\u8def\u5f84\u7aef\u70b9\u7684\u7b54\u6848\u6570\u91cf\u2014\u2014\u8fd9\u79cd\u7b54\u6848\u662f\u4e0a\u8ff0\u65b9\u6cd5\u672a\u7edf\u8ba1\u5230\u7684\u3002\u6240\u4ee5\uff0c\u6211\u4eec\u9700\u8981\u5c06\u6839\u8282\u70b9\u7684\u7b54\u6848\u51cf\u53bb\\(val_{a_{rt}}\\)\uff0c\u7136\u540e\u518d\u52a0\u4e0a\u4ee5\u6839\u8282\u70b9\u4e3a\u6839\u7684\u6240\u6709\u8def\u5f84\u6570\u91cf\uff0c\u4e5f\u5c31\u662f\\(sz_{rt}\\)<\/p>\n\n\n\n<pre class=\"wp-block-code\"><code>#include&lt;iostream>\n#include&lt;cstdio>\n#include&lt;set>\n#define Fv(i,X) for(int i=h[X];i;i=e[i].nxt)\n\ntypedef long long ll;\n\ninline int Max(int A,int B){\n\treturn A>B?A:B;\n}\ninline ll Max(ll A,ll B){\n\treturn A>B?A:B;\n}\n\nstruct ee{\n\tint v;\n\tint nxt;\n}e[200005];\nint h[100005],et=0;\ninline void Eadd(int U,int V){\n\te[++et]=(ee){V,h[U]};\n\th[U]=et;\n}\ninline void add(int U,int V){\n\tEadd(U,V);\n\tEadd(V,U);\n}\n\nint n,a[100005];\nint s,rt=0;\nint vis[100005];\nll sz[100005],mx[100005],ans[100005];\ninline void dfs0(int X,int FA){\n\tsz[X]=1,mx[X]=0;\n\tFv(i,X){\n\t\tif(e[i].v==FA||vis[e[i].v]){\n\t\t\tcontinue;\n\t\t}\n\t\tdfs0(e[i].v,X);\n\t\tsz[X]+=sz[e[i].v];\n\t\tmx[X]=Max(mx[X],sz[e[i].v]);\n\t}\n\tmx[X]=Max(mx[X],s-sz[X]);\n\tif(mx[X]&lt;mx[rt]){\n\t\trt=X;\n\t}\n}\nll cnt[100005],val[100005],sm=0,nm=0,dlt;\ninline void dfs2(int X,int FA){\n\tsz[X]=1;\n\t++cnt[a[X]];\n\tFv(i,X){\n\t\tif(e[i].v==FA||vis[e[i].v]){\n\t\t\tcontinue;\n\t\t}\n\t\tdfs2(e[i].v,X);\n\t\tsz[X]+=sz[e[i].v];\n\t}\n\tif(cnt[a[X]]==1){\n\t\tsm+=sz[X];\n\t\tval[a[X]]+=sz[X];\n\t}\n\t--cnt[a[X]];\n}\ninline void dfs3(int X,int FA,int TYP){\n\t++cnt[a[X]];\n\tFv(i,X){\n\t\tif(e[i].v==FA||vis[e[i].v]){\n\t\t\tcontinue;\n\t\t}\n\t\tdfs3(e[i].v,X,TYP);\n\t}\n\tif(cnt[a[X]]==1){\n\t\tsm+=sz[X]*TYP;\n\t\tval[a[X]]+=sz[X]*TYP;\n\t}\n\t--cnt[a[X]];\n}\ninline void dfs4(int X,int FA){\n\t++cnt[a[X]];\n\tif(cnt[a[X]]==1){\n\t\t++nm;\n\t\tsm-=val[a[X]];\n\t}\n\tans[X]+=sm+nm*dlt;\n\tFv(i,X){\n\t\tif(e[i].v==FA||vis[e[i].v]){\n\t\t\tcontinue;\n\t\t}\n\t\tdfs4(e[i].v,X);\n\t}\n\tif(cnt[a[X]]==1){\n\t\t--nm;\n\t\tsm+=val[a[X]]; \n\t}\n\t--cnt[a[X]];\n}\ninline void dfs5(int X,int FA){\n\tval[a[X]]=cnt[a[X]]=0;\n\tFv(i,X){\n\t\tif(e[i].v==FA||vis[e[i].v]){\n\t\t\tcontinue;\n\t\t}\n\t\tdfs5(e[i].v,X);\n\t}\n}\ninline void calc(int X){\n\tsm=nm=0;\n\tdfs2(X,0);\n\tans[rt]+=sm-val[a[X]]+sz[X];\n\tFv(i,X){\n\t\tif(vis[e[i].v]){\n\t\t\tcontinue;\n\t\t}\n\t\t++cnt[a[X]];\n\t\tsm-=sz[e[i].v];\n\t\tval[a[X]]-=sz[e[i].v];\n\t\tdfs3(e[i].v,X,-1);\n\t\t--cnt[a[X]];\n\t\t\n\t\tdlt=sz[X]-sz[e[i].v];\n\t\tdfs4(e[i].v,X);\n\t\t\n\t\t++cnt[a[X]];\n\t\tsm+=sz[e[i].v];\n\t\tval[a[X]]+=sz[e[i].v]; \n\t\tdfs3(e[i].v,X,1);\n\t\t--cnt[a[X]];\n\t}\n\tdfs5(X,0);\n}\n\ninline void dfs1(int X){\n\tvis[X]=1;\n\tcalc(X);\n\tFv(i,X){\n\t\tif(vis[e[i].v]){\n\t\t\tcontinue;\n\t\t}\n\t\ts=sz[X],rt=0;\n\t\tdfs0(e[i].v,X);\n\t\tdfs1(rt);\n\t}\n}\n\nvoid init(){\n\tscanf(\"%d\",&amp;n);\n\tfor(int i=1;i&lt;=n;++i){\n\t\tscanf(\"%d\",&amp;a[i]);\n\t}\n\tint u,v;\n\tfor(int i=1;i&lt;n;++i){\n\t\tscanf(\"%d%d\",&amp;u,&amp;v);\n\t\tadd(u,v);\n\t}\n\trt=0,mx[0]=n;\n\tdfs0(1,0);\n\tdfs1(rt);\n\tfor(int i=1;i&lt;=n;++i){\n\t\tprintf(\"%lld\\n\",ans[i]);\n\t}\n}\n\nint main(){\n\tinit();\n\treturn 0;\n}<\/code><\/pre>\n","protected":false},"excerpt":{"rendered":"<p>\u4e00\u79cd\\(n^2log^2n\\)\u7684\u505a\u6cd5\u662f\u53ef\u4ee5\u60f3\u5230\u7684\uff1a\u70b9\u5206\u6cbb\uff0c\u4f7f\u7528multiset\u6765\u7ef4\u62a4\u4ee5\u67d0\u4e2a\u8282\u70b9\u4e3a\u6839\u7684\u67d0\u4e00\u4e9b\u94fe\u4e0a &hellip; <\/p>\n<p class=\"link-more\"><a href=\"http:\/\/SmokeyDays.top\/wordpress\/2019\/03\/28\/lp2664-%e6%a0%91%e4%b8%8a%e6%b8%b8%e6%88%8f\/\" class=\"more-link\">\u7ee7\u7eed\u9605\u8bfb<span class=\"screen-reader-text\">\u201clp2664 \u6811\u4e0a\u6e38\u620f\u201d<\/span><\/a><\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":[],"categories":[30,56,8,105,6,5],"tags":[],"_links":{"self":[{"href":"http:\/\/SmokeyDays.top\/wordpress\/wp-json\/wp\/v2\/posts\/837"}],"collection":[{"href":"http:\/\/SmokeyDays.top\/wordpress\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"http:\/\/SmokeyDays.top\/wordpress\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"http:\/\/SmokeyDays.top\/wordpress\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"http:\/\/SmokeyDays.top\/wordpress\/wp-json\/wp\/v2\/comments?post=837"}],"version-history":[{"count":1,"href":"http:\/\/SmokeyDays.top\/wordpress\/wp-json\/wp\/v2\/posts\/837\/revisions"}],"predecessor-version":[{"id":838,"href":"http:\/\/SmokeyDays.top\/wordpress\/wp-json\/wp\/v2\/posts\/837\/revisions\/838"}],"wp:attachment":[{"href":"http:\/\/SmokeyDays.top\/wordpress\/wp-json\/wp\/v2\/media?parent=837"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"http:\/\/SmokeyDays.top\/wordpress\/wp-json\/wp\/v2\/categories?post=837"},{"taxonomy":"post_tag","embeddable":true,"href":"http:\/\/SmokeyDays.top\/wordpress\/wp-json\/wp\/v2\/tags?post=837"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}