第一题过于简单就不说了。
第二题:合并石子,看题目以为是个区间dp
但是其实不难发现性质:切割之后的最终态,每一堆都一定匹配一个l1,r1于是可以双指针,每匹配到相同的数就分出来即可。
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int a[N];
int b[N];
int n,m;
int main()
{
cin >> n >> m;
for(int i = 1 ; i <= n ; i ++) cin >> a[i];
for(int i = 1 ; i <= m ; i ++) cin >> b[i];
int l1 = 1 , r1 = n;
int l2 = 1, r2 = m;
int cnt = 0;
while(l1 <= r1 && l2 <= r2)
{
long long sum1 = a[l1],sum2 = b[l2];
int az = l2;
int aq = l1;
while(sum1 != sum2 && aq <= r1 && az <= r2 )
{
if(sum1 < sum2) sum1 += a[++aq];
else if(sum2 < sum1) sum2 += b[++az];
}
if(sum1 == sum2) l1 = ++aq,l2 = ++az,cnt++;
}
cout << cnt << endl;
}
第三题:翻转树边
换根dp板子题,关键问题是如何建图,这里我选择用边权建图,0表示 x 到y 有一条边,
1 表示y 到x 有一条边 ,接着一次dfs便可以求出根节点的对应值,接下来手推个公式
会发现 : 不妨设当前节点为u, 儿子节点为Y,F(x)记录为对应翻转的边数,于是有
如果父亲指向儿子节点的边权值是1
如果父亲指向儿子节点的边权值是0
接着遍历打印出答案即可。(这题手机上打的,感谢荣巨帮忙提交)
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 200000 + 10, M = 2 * N;
int h[N],e[M],ne[M],idx,edge[M];
//0 表示x到y , 1表示y到x
int n;
int d[N];
int f[N];
void add (int a,int b,int c)
{
e[idx]=b;
ne[idx]=h[a];
edge[idx] = c;
h[a]=idx++;
}
void dfs1(int u,int fa)
{
for(int i = h[u] ; i != - 1 ; i = ne[i])
{
int y = e[i];
if(y == fa)continue;
dfs1(y,u);
if(edge[i] == 1) d[u] += d[y] + 1;
else d[u] += d[y];
}
}
void dfs2(int u,int fa)
{
for(int i = h[u] ; i != - 1 ; i = ne[i])
{
int y = e[i];
if(y == fa)continue;
// dfs2(y,u);
if(edge[i] == 1) f[y] = f[u] - 1;
else f[y] = f[u] + 1 ;
dfs2(y,u);
}
}
int main()
{
memset(h,-1,sizeof h);
cin >> n;
for(int i = 1 ; i < n ; i++)
{
int x,y;
cin >> x >> y;
add(x,y,0);
add(y,x,1);
}
dfs1(1,0);
f[1] = d[1];
dfs2(1,0);
int maxx = 0x3f3f3f3f;
for(int i = 1; i <= n ; i++) if(f[i] < maxx) maxx = f[i];
cout << maxx << endl;
// cout << f[1] <<" "<<f[2] <<" "<<f[3] << endl;
for(int i = 1 ; i <= n ; i++)
{
if(f[i] == maxx) cout << i <<" ";
}
}