Submission #18559763

---

Source Code Expand

#import <Foundation/Foundation.h>

int main(int argc, const char * argv[]) {
    @autoreleasepool {

        NSData *stdinData = [[NSFileHandle fileHandleWithStandardInput] availableData];
        NSString *stdinString = [[NSString alloc] initWithData:stdinData encoding:NSUTF8StringEncoding];
        NSMutableArray<NSMutableArray<NSString *> *> *stdinStrArr = [NSMutableArray new];
        for (NSString *str in [stdinString componentsSeparatedByString:@"\n"]) {
            [stdinStrArr addObject:[[str componentsSeparatedByString:@" "] mutableCopy]];
        }

        NSMutableArray<NSNumber *> *aList = [NSMutableArray new];
        for (NSString *i in [stdinStrArr objectAtIndex:1]) {
            NSNumber *currentIndex = [NSNumber numberWithLongLong:[i longLongValue]];
            long long replaceIndex = [currentIndex longLongValue] - 1;
            [aList addObject:[NSNumber numberWithLongLong:0]];
            NSNumber *newValue = [NSNumber numberWithLongLong:[[aList objectAtIndex:replaceIndex] longLongValue] + 1];
            [aList replaceObjectAtIndex:replaceIndex withObject:newValue];
        }
        [aList addObject:[NSNumber numberWithLongLong:0]];

        NSFileHandle *fileHandle = [NSFileHandle fileHandleWithStandardOutput];

        for (NSNumber *n in aList) {
            [fileHandle writeData:[[NSString stringWithFormat:@"%@", n] dataUsingEncoding:NSUTF8StringEncoding]];
            [fileHandle writeData:[@"\n" dataUsingEncoding:NSUTF8StringEncoding]];
        }
        [fileHandle closeFile];

    }
    return 0;
}

Submission Info

Judge Result