這個題其實可以轉化成在一些數字之間插上加號或者減號,然後等式的計算結果是t,求方案的一個題。其中第一個數和第二個數的符號是固定的:+,-
f[i][j]表示前i個數的運算結果是j時,第i個數前面的符號。所以如果存在f[i][j-a[i]],則可以轉移到f[i][j]=1上。同理如果存在f[i][j+a[i]],則可以轉移到f[i][j]=0上。這樣我們就構造好了這個數組。
然後開始找出對於要求答案每一位是+還是-。從後往前找,f[n][t]的符號就是最後一位前的符號。因爲知道第n位的正負,然後我們由此就可以知道第n-1位的符號。以此類推,我們可以知道每一位前的符號。
現在問題就成了如何從知道每一位的符號然後知道每一次改的是哪2個數。
碰到一個加號就把當前2個數(i,i-1)相減。處理完所有的加號之後我們就可以一直輸出1了。
#include <bits/stdc++.h>
using namespace std;
const int maxn=105;
int T=10000;
int n,t,a[maxn],f[maxn][20005];
int opt[maxn];
int main()
{
scanf("%d%d",&n,&t);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
memset(f,-1,sizeof(f));
//T是爲了防止數組是負的
f[1][a[1]+T]=1;//+
f[2][a[1]-a[2]+T]=0;//-
for(int i=3;i<=n;i++)
{
for(int j=0;j<=2*T;j++)
{
if(f[i-1][j]!=-1)
{
f[i][j+a[i]]=1;
f[i][j-a[i]]=0;
}
}
}
int j=T+t;
for(int i=n;i>=1;i--)
{
if(f[i][j]==1)
{
opt[i]=1;
j-=a[i];
}
else if(f[i][j]==0)
{
opt[i]=0;
j+=a[i];
}
}
int cnt=0;
for(int i=2;i<=n;i++)
{
if(opt[i]==1)
{
printf("%d\n",i-cnt-1);
cnt++;
}
}
for(int i=2;i<=n;i++) if(opt[i]==0) printf("1\n");
return 0;
}