呵呵呵呵呵呵，自己画图，大概半个小时，觉的连上边会成环（是不是该交仙人掌了？？）然后求环不重合部分最大就好了，
结果写了一坨DP，最后写不下去了，再次扒了题解。
发现我真的是个sb。
k==1，直接是直径
k==2，搞出直径然后把直径删掉（把权值赋为-1,再找直径）（有点像我一开始想的每次找个最长链去贪心，然而，，总觉得，这种题贪心这么可能对）
1 /*#include <bits/stdc++.h>
2 #define LL long long
3 #define lowbit(x) x&(-x)
4 #define inf 0x3f3f3f3f
5 #define N 100005
6 using namespace std;
7 inline int ra()
8 {
9 int x=0,f=1; char ch=getchar();
10 while (ch<‘0‘ || ch>‘9‘) {if (ch==‘-‘) f=-1; ch=getchar();}
11 while (ch>=‘0‘ && ch<=‘9‘) {x=x*10+ch-‘0‘; ch=getchar();}
12 return x*f;
13 }
14 int head[N],cnt,n,f[N][3][3],k;
15 struct edge{int next,to;}e[N<<1];
16 void insert(int x, int y){e[++cnt].next=head[x]; e[cnt].to=y; head[x]=cnt;}
17 void dfs(int x, int fa)
18 {
19 int mx1=0,mx2=0,mx3=0,mx4=0;
20 for (int i=head[x];i;i=e[i].next)
21 {
22 if (e[i].to==fa) continue;
23 dfs(e[i].to,x);
24 int orz=f[e[i].to][0][1],flag=1,hehe=1;
25 if (orz<=mx1 && orz<=mx2 && orz<=mx3 && orz>mx4) mx4=orz;
26 if (orz<=mx1 && orz<=mx2 && orz>mx3) mx4=mx3,mx3=orz;
27 if (orz<=mx1 && orz>mx2) mx4=mx3,mx3=mx2,mx2=orz,hehe=0;
28 if (orz>mx1) mx4=mx3,mx3=mx2,mx2=mx1,mx1=orz,flag=0;
29 if (f[x][1][1] && k)
30 {
31 if (!flag)
32 f[x][2][0]=max(f[x][2][0],f[x][1][1]+mx1+1);
33 else if (!hehe) f[x][2][0]=max(f[x][2][0],f[x][1][1]+mx2+1);
34 }
35 cout<<f[x][1][1]<<endl;
36 if (f[x][1][0] && k) f[x][1][1]=max(f[x][1][1],f[x][1][0]+mx1+1);
37 cout<<f[x][1][1]<<endl;
38 if (f[e[i].to][1][0] && mx2 && k) f[x][1][1]=max(f[x][1][1],1+f[e[i].to][1][0]+flag?mx1:mx2);
39 if (f[e[i].to][1][1] && k) f[x][1][1]=max(f[x][1][1],f[e[i].to][1][1]+1);
40 if (f[e[i].to][1][1] && mx2 && k) f[x][2][0]=max(f[x][2][0],1+f[e[i].to][1][1]+flag?mx1:mx2);
41 if (f[e[i].to][2][0] && k) f[x][2][0]=max(f[x][2][0],f[e[i].to][2][0]);
42 if (f[x][1][0] && f[e[i].to][1][0] && k) f[x][2][0]=max(f[x][2][0],f[x][1][0]+f[e[i].to][1][0]);
43 f[x][1][0]=max(f[x][1][0],max(mx1+mx2+1,f[e[i].to][1][0]));
44 f[x][0][1]=mx1+1;
45 if (f[3][1][1]==3) {cout<<x<<" "<<e[i].to; while (1);}
46 }
47 printf("%d %d %d %d %d\n",x,f[x][0][1],f[x][1][0],f[x][1][1],f[x][2][0]); system("pause");
48 }
49 int main(int argc, char const *argv[])
50 {
51 n=ra(); k=ra(); k--;
52 for (int i=1; i<n; i++)
53 {
54 int x=ra(),y=ra();
55 insert(x,y); insert(y,x);
56 }
57 dfs(1,0);
58 printf("%d\n",n*2-f[1][k+1][0]);
59 return 0;
60 }*/
61 #include <bits/stdc++.h>
62 #define LL long long
63 #define lowbit(x) x&(-x)
64 #define inf 0x3f3f3f3f
65 #define N 100005
66 using namespace std;
67 inline int ra()
68 {
69 int x=0,f=1; char ch=getchar();
70 while (ch<‘0‘ || ch>‘9‘) {if (ch==‘-‘) f=-1; ch=getchar();}
71 while (ch>=‘0‘ && ch<=‘9‘) {x=x*10+ch-‘0‘; ch=getchar();}
72 return x*f;
73 }
74 int mx,n,k,tot,cnt=1,len;
75 int head[N];
76 int s1[N],s2[N];
77 struct edge{
78 int to,next,v;
79 }e[N<<1];
80 void insert(int x, int y){
81 e[++cnt].next=head[x]; e[cnt].to=y; e[cnt].v=1; head[x]=cnt;
82 }
83 int dfs(int x, int fa)
84 {
85 int mx1=0,mx2=0;
86 for (int i=head[x];i;i=e[i].next)
87 {
88 if (e[i].to==fa) continue;
89 int v=e[i].v+dfs(e[i].to,x);
90 if (v>mx1) mx2=mx1,mx1=v,s2[x]=s1[x],s1[x]=i;
91 else if (v>mx2) mx2=v,s2[x]=i;
92 }
93 if (mx1+mx2>len) len=mx1+mx2,mx=x;
94 return mx1;
95 }
96 int main(int argc, char const *argv[])
97 {
98 n=ra(); k=ra(); tot=2*(n-1);
99 for (int i=1; i<n; i++)
100 {
101 int x=ra(),y=ra();
102 insert(x,y);
103 insert(y,x);
104 }
105 dfs(1,0); tot-=len-1;
106 if (k==2)
107 {
108 len=0;
109 for (int i=s1[mx];i;i=s1[e[i].to]) e[i].v=e[i^1].v=-1;
110 for (int i=s2[mx];i;i=s1[e[i].to]) e[i].v=e[i^1].v=-1;
111 dfs(1,0); tot-=len-1;
112 }
113 cout<<tot;
114 return 0;
115 }
时间： 2024-09-29 22:21:27