砝码称重
题目链接
题意:你有一架天平和 N 个砝码,这 N 个砝码重量依次
W
1
,
W
2
,
⋅
⋅
⋅
,
W
n
W1,W2,⋅⋅⋅,Wn
W1,W2,⋅⋅⋅,Wn。请你计算一共可以称出多少种不同的正整数重量?注意砝码可以放在天平两边。
对于 50% 的评测用例,1≤N≤15。
对于所有评测用例,1≤N≤100,N 个砝码总重不超过
1
0
5
10^5
105。
思路:可以将题目转化为某个重量能否被称量,然后进行记忆化搜索。
#include <iostream>
#include <cstring>
using namespace std;
const int base=1e5;
int n,a[107];
int dp[107][200007];
int dfs(int p,int s){
if(p>n){
if(s==base) return 1;
return 0;
}
if(dp[p][s]!=-1) return dp[p][s];
int ok=0;
if(!ok) ok=dfs(p+1,s-a[p]);
if(!ok) ok=dfs(p+1,s+a[p]);
if(!ok) ok=dfs(p+1,s);
return dp[p][s]=ok;
}
int main(){
cin>>n;
memset(dp,-1,sizeof(dp));
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
int sum=0;
for(int i=1;i<=100000;i++){
if(dfs(1,i+base)) sum++;
}
cout<<sum<<endl;
}
D. Armchairs
题目链接
题意:将每个初始为1的点放到初始为0的点最少需要多长时间,每个1不可同时移动,每个0只能放一个1。
思路:将1和0的位置分别存储在两个数组中,进行记忆化搜索,从而得到最小值。
#include<bits/stdc++.h>
using namespace std;
#define ll long long
vector<int>v1,v2;
int ans1,ans2;
int dp[5010][5010];
int dfs(int pos1,int pos2){
if(pos1==ans2){
if(pos2==ans1){
return 0;
}
else return 1e9;
}
if(dp[pos1][pos2]!=-1)return dp[pos1][pos2];
int ans=1e9;
if(pos2<=ans1-1){
ans=min(ans,dfs(pos1+1,pos2+1)+abs(v2[pos1]-v1[pos2]));
}
ans=min(ans,dfs(pos1+1,pos2));
return dp[pos1][pos2]=ans;
}
int main(){
int n;
scanf("%d",&n);
memset(dp,-1,sizeof(dp));
int x;
for(int i=1;i<=n;i++){
scanf("%d",&x);
if(x==1){
v1.push_back(i);
}
else v2.push_back(i);
}
ans1=v1.size();
ans2=v2.size();
int f=dfs(0,0);
printf("%d\n",f);
}