51nod 1624 取余最长路[set]

注意到只有三行,两个拐弯的位置
假设拐弯的位置为$x,y$
则$ans=(sum[3][n]-sum[3][y-1])+(sum[2][y]-sum[2][x-1])+sum[1][x]=(sum[3][n]-sum[3][y-1])+sum[2][y]+(sum[1][x]-sum[2][x-1])$
于是和$x,y$有关的量分别处理,$(sum[1][x]-sum[2][x-1])$塞到set里枚举$y$就好了
统计答案时要使$SUM$尽量接近$p$(到set里找第一个小于$p-((sum[3][n]-sum[3][y-1])+sum[2][y])$的值,注意考虑所有数都比它大的情况
题比较简单,主要是练习一下stl

#include<cstdio>
#include<algorithm>
#include<set>
#define N 100000
#define LL long long
using namespace std;

int n,p,pre[4][N+5];
LL ans;

set<LL>s;

int main() {
    scanf("%d %d",&n,&p);
    for(int i=1;i<=3;++i)
        for(int j=1,x;j<=n;++j) {
            scanf("%d",&x);
            if(i<=2) (pre[i][j]=pre[i][j-1]+x)%=p;
            else pre[i][j]=x;
        }
    for(int i=n;i>=1;--i)
        (pre[3][i]+=pre[3][i+1])%=p;
    ans=0;
    for(int i=1;i<=n;++i) {
        LL tmp;
        tmp=(pre[1][i]-pre[2][i-1]+p)%p;
        s.insert(tmp);
        tmp=(pre[3][i]+pre[2][i])%p;
        set<LL>::iterator it;
        it=s.lower_bound(p-tmp);
        if(it==s.begin()) ans=max(ans,(tmp+(*(--s.end())))%p);
        else ans=max(ans,(tmp+(*(--it)))%p);
    }
    printf("%lld\n",ans);
}